Sealbook

patrick-kidger/diffrax

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

deep-learningdifferential-equationsdynamical-systemsequinoxjaxmachine-learningneural-differential-equationsneural-networks

License & maintenance

Apache-2.0 is a permissive license: free to use commercially with attribution.

Last code push 43 days ago.