Jax (and leapfrog integration) for differential equation approximate solutions?

Questions
2

Any reason you wouldn’t want to solve the ODE directly with JAX, like this?

FWIW, that Op should work for any JAX-friendly function, it doesn’t have to be an ODE.

1 Like