google-deepmind/optax

optax

✅ Recently maintained

Optax is a gradient processing and optimization library for JAX.

2,360Stars

388Forks

+17-day stars

Python · Apache-2.0 · Last push 2026-10-30

Star history

2025-09-292026-11-02

View historical data

git clone https://github.com/google-deepmind/optax.git

Related domains do not imply feature parity. Review official documentation and licenses.

Google's open source machine learning framework for building and deploying deep learning models at any scale.

Leading open source machine learning framework with dynamic execution.

12 weeks, 26 lessons, 52 quizzes, classic Machine Learning for all