
optax
by google-deepmind (organization)
Optax is a gradient processing and optimization library for JAX. (description from the project’s GitHub page)
Repository facts
- Stars
- 2,351
- Forks
- 379
- Open issues and pull requests
- 144
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- google-deepmind · Organization
- Created
- Jun 12, 2020
- Last push
- Oct 9, 2026
Topics
Add this badge to your README
Shows this repository’s rank in AI on GitHub and links back to this page. Ranks are by stars and change when the list is refreshed. The badge is not an endorsement or a rating. About the badges
Markdown (for a README)
[](https://unitedstatesofamericasuperintelligence.com/github/google-deepmind--optax/)
HTML
<a href="https://unitedstatesofamericasuperintelligence.com/github/google-deepmind--optax/"><img src="https://unitedstatesofamericasuperintelligence.com/badges/github/google-deepmind--optax.svg" alt="Ranked #3,212 in AI on GitHub" height="20"></a>
Related projects
Projects that share this one’s GitHub topics.
Rank 138: ray-project/ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
44,061 starsPythonApache-2.0Last push Oct 18, 2026
Rank 2,237: PennyLaneAI/pennylane
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiratio…
3,501 starsPythonApache-2.0Last push Oct 17, 2026
Rank 3,357: ahmedfgad/GeneticAlgorithmPython
Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).
2,232 starsPythonBSD-3-ClauseLast push Oct 17, 2026
Rank 4,211: google/vizier
Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
1,680 starsPythonApache-2.0Last push Oct 14, 2026
Rank 4,428: tensorflow/model-optimization
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
1,580 starsPythonApache-2.0Last push Oct 14, 2026
Rank 4,524: AxeldeRomblay/MLBox
MLBox is a powerful Automated Machine Learning python library.
1,534 starsPythonOther licenseLast push Aug 6, 2023