Sign in
← Back to search

patrick-kidger/equinox

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

#deep-learning#equinox#jax#neural-networks
Sign in to like patrick-kidger/equinox Open GitHub
Stars
2,888
Forks
197
Commits
1126
Language
Python
Awesome lists
1

Similar repositories

patrick-kidger/diffrax

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

2013 stars
Python 1 awesome list

jax-ml/jax

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

35740 stars
Python 3 awesome lists

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

2272 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

facebook/Ax

Adaptive Experimentation Platform

2758 stars
Python 1 awesome list

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-29
  • First commit: —
  • Last pushed: 2026-05-22
  • Archived: no
  • Stack detected: —
  • License: Apache-2.0

AI development signals

No AI development config files detected.