Implement the IAS15 solver

patrick-kidger / diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Apache License 2.0

1.42k stars 127 forks source link

Implement the IAS15 solver #437

Open nstarman opened 4 months ago

nstarman commented 4 months ago

https://ui.adsabs.harvard.edu/abs/2015MNRAS.446.1424R/abstract It's an adaptive symplectic integrator with superior speed and performance. Currently the symplectic solver in diffrax is not adaptive. Implementing this solver would offer an improved symplectic option.

patrick-kidger commented 4 months ago

Sounds good! I'd be happy to consider a PR on this.