Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Primary LanguagePythonApache License 2.0Apache-2.0
No one’s watching this repository yet.