Sign in
← Back to search

patrick-kidger/diffrax

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

#deep-learning#differential-equations#dynamical-systems#equinox#jax#machine-learning#neural-differential-equations#neural-networks
Sign in to like patrick-kidger/diffrax Open GitHub
Stars
2,013
Forks
181
Commits
727
Language
Python
Awesome lists
1

Similar repositories

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

2888 stars
Python 1 awesome list

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

35728 stars
Python 3 awesome lists

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

2272 stars
Python 1 awesome list

pbrod/numdifftools

Solve automatic numerical differentiation problems in one or more variables.

281 stars
Python 1 awesome list

google/flax

Flax is a neural network library for JAX that is designed for flexibility.

7206 stars
Jupyter Notebook 2 awesome lists

Tracked growth

1 capture since 2026-05-25

Latest capture 2026-05-25 21:14

Stars history

Total stars

Commits history

Default branch commits

Metadata

  • Created: 2021-07-25
  • First commit: —
  • Last pushed: 2026-05-13
  • Archived: no
  • Stack detected: —
  • License: Apache-2.0

AI development signals

No AI development config files detected.