patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
License & maintenance
Apache-2.0 is a permissive license: free to use commercially with attribution.
Last code push 43 days ago.