in kbc/models.py [0:0]
def get_rhs(self, chunk_begin: int, chunk_size: int): return self.rhs.weight.data[ chunk_begin:chunk_begin + chunk_size ].transpose(0, 1)