Can’t you make your potential accept higher dimensionality data? Then if things have the same shapes you could just pass the whole block of parameters instead of slicing it 50 times.
Can’t you make your potential accept higher dimensionality data? Then if things have the same shapes you could just pass the whole block of parameters instead of slicing it 50 times.