optax

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

• machine-learning
• optimization

Rank #3,212 of 9,507 in AI on GitHub, by stars. Data from the GitHub API, fetched Oct 18, 2026; star counts change daily. Stars measure popularity, not quality or safety, and a listing is not an endorsement. How this list is made

Related projects

Projects that share this one’s GitHub topics.

• 138

  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

• 2,237

  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

• 3,357

  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

• 4,211

  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

• 4,428

  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

• 4,524

  Rank 4,524: AxeldeRomblay/MLBox

  MLBox is a powerful Automated Machine Learning python library.

  1,534 starsPythonOther licenseLast push Aug 6, 2023