automl21/scs_neural/solver/linear_operator.py (1 line):
	- line 98: # TODO: make sure it batches correctly