patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2888 stars
Python
1 awesome list
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
- Stars
- 2,272
- Forks
- 328
- Commits
- 1976
- Language
- Python
- Awesome lists
- 1
Similar repositories
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
35728 stars
Python
3 awesome lists
facebook/Ax
Adaptive Experimentation Platform
2758 stars
Python
1 awesome list
axolotl-ai-cloud/axolotl
Go ahead and axolotl questions
11960 stars
Python
2 awesome lists
google-deepmind/dm-haiku
JAX-based neural network library
3234 stars
Python
1 awesome list
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
2013 stars
Python
1 awesome list
Tracked growth
3 captures since 2026-05-22
Latest capture 2026-05-29 03:00
Stars history
Total starsCommits history
Default branch commitsMetadata
- Created: 2020-06-12
- First commit: 2020-07-09
- Last pushed: 2026-05-29
- Website: https://optax.readthedocs.io
- Archived: no
- Stack detected: —
- License: Apache-2.0
AI development signals
No AI development config files detected.