Parameter tuning in neural ODE

BS3 at low tolerance is going to be slow. I’d suggest a different integrator. I may be able to look into this on the weekend.