jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
35728 stars
Python
3 awesome lists
- Stars
- 7,206
- Forks
- 810
- Commits
- 5860
- Language
- Jupyter Notebook
- Awesome lists
- 2
Similar repositories
google-deepmind/dm-haiku
JAX-based neural network library
3234 stars
Python
1 awesome list
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2888 stars
Python
1 awesome list
axolotl-ai-cloud/axolotl
Go ahead and axolotl questions
11960 stars
Python
2 awesome lists
microsoft/FLAML
A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.
4354 stars
Jupyter Notebook
1 awesome list
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
2272 stars
Python
1 awesome list
Tracked growth
3 captures since 2026-05-22
Latest capture 2026-05-25 20:58
Stars history
Total starsCommits history
Default branch commitsMetadata
- Created: 2020-01-10
- First commit: —
- Last pushed: 2026-05-22
- Website: https://flax.readthedocs.io
- Archived: no
- Stack detected: —
- License: Apache-2.0
AI development signals
No AI development config files detected.