patrick-kidger/diffrax

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

Training & fine-tuningPython Apache-2.0 204created 2021-07-25

Stars

2,124

1-day growth

—

7-day growth

—

30-day growth

—

Star history

Tracked since 2026-11-13

#deep-learning#differential-equations#dynamical-systems#equinox#jax#machine-learning#neural-differential-equations#neural-networks

Top contributors

Not collected yet (contributors job).