diffrax
diffrax copied to clipboard
ConstantStepSize(compile_steps=True) w/ SaveAt(steps=True) allocates an overly-large output array
In this case the size of the output array should be equal to the known-at-compile-time number of steps.