ODE Solver Speed-up Suggestion

[adam-hartshorne]

With the growing popularity of Neural ODEs, Jax currently experimental ODE is a bit slow. Thought it is worth pointing out this,

One simple-to-implement trick dramatically improves the speed at which Neural ODEs and Neural CDEs can be trained. (As much as doubling the speed.)

https://github.com/patrick-kidger/FasterNeuralDiffEq