optax
by google-deepmind · GitHub
Optax is a gradient processing and optimization library for JAX.
# machine-learning# optimization
⭐ Stars
2.3k
🍴 Forks
346
🔥 Trending
+2 this week
📜 License
Apache-2.0
Commercial use OK📅 Created
2020
🔄 Last commit
6 days ago
🏷️ Category
machine-learning
💻 Language
You maintain this project?
Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.
📈 Star history
2026-07-072026-07-22
📄 About
Optax is a gradient processing and optimization library for JAX.
❓ Frequently asked questions
What is optax?
Optax is a gradient processing and optimization library for JAX.
Is optax open source?
optax is an open-source project. It is released under the Apache-2.0 license.
Is optax free?
Yes. optax is free and open source — you can use, modify and self-host it.
🏅 Maintainer of this project?
Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.
[](https://opensourceai.tech/project/google-deepmind-optax.html)
🧬 Shares DNA with🧬 View the DNA map →
robotics_master
Online Robotics Master
★ 350 · computer-vision
sharesoptimization
model-based-diffusion
Official implementation for the paper "Model-based Diffusion for Trajectory Optimization". Mode…
★ 339 · control
sharesoptimization
AFlow
🔥🔥🔥 ICLR 2025 Oral. Automating Agentic Workflow Generation.
★ 554 · agentic-workflow
sharesoptimization
Measured from GitHub topics shared by both projects, weighted by how rare each topic is.