google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
Star History
Add Live Star History & Verified Badges to README.md
Keep your repository README looking professional and dynamic. As our continuous crawler records new stars, these official SVG badges update in real time with zero maintenance.
Renders your high-resolution star trajectory chart right inside your GitHub README or project docs.
[](https://githubrepo.cloud/repo/google-deepmind/optax?utm_source=readme_chart)Compact Shields-style badges for your README header. Shows real-time stars and global ranking.
[](https://githubrepo.cloud/repo/google-deepmind/optax?utm_source=readme_badge) [](https://githubrepo.cloud/repo/google-deepmind/optax?utm_source=readme_badge) [](https://githubrepo.cloud/repo/google-deepmind/optax?utm_source=readme_badge)Momentum
+59
STARS · LAST 30 DAYS
2
PER DAY
#350
MOST-STARRED Python
| Window | 7 days | 30 days | 90 days |
|---|---|---|---|
| Stars gained | +14 | +59 | +180 |
| Per day | 2 | 2 | 2 |
| Forks gained | +2 | +7 | +19 |
optax gained 59 stars in the last 30 days, about 2 a day, and now has 2.3K. It is about 6 years old and has averaged roughly 391 stars a year. It ranks #350 among Python repositories and #2,426 across all languages on GitHubRepo.
Trending Record
1
DAYS ON TRENDING
#3276
BEST RANK
Oct 1, 2026
FIRST APPEARANCE
Active
STATUS TODAY
optax has maintained a continuous presence across global trending indexes, peaking at #3276. Below is the 30-day activity profile:
💡 Overview
optax is an open-source project written in Python: Optax is a gradient processing and optimization library for JAX.
Engineered for speed, consistency, and developer ease, it solves common hurdles in machine-learning, optimization. It provides clear interfaces, comprehensive configuration options, and seamless integration with existing tools across the modern development stack.
⚡ Key Features
Optimized execution pipeline written in Python for predictable speed.
Zero-friction configuration with comprehensive sensible defaults out of the box.
Cross-platform runtime support across Linux, macOS, and Windows environments.
Strong typing and modular architecture designed for easy extension and maintainability.
Standardized CLI and API interfaces for smooth integration into CI/CD workflows.
Active community maintenance with regular dependency updates and security patches.
📥 Installation
$ pip install optax⚙ System Requirements
Platforms
- • macOS
- • Linux
- • Windows
Runtime & Dependencies
Python >= 3.9, pip, virtualenv
Architecture
x86_64, ARM64 (Apple Silicon & Graviton)
🧠 How It Works
optax coordinates its core functionality through a modular Python pipeline. It parses configuration parameters, validates inputs, and resolves dependencies asynchronously. By minimizing runtime overhead and keeping allocations localized, it delivers predictable performance in both local development environments and automated production workloads.
🎯 Production Use Cases
Production System Integration
Embed optax into Python backend services to handle core application logic.
CI/CD Automated Pipelines
Run automated validation, builds, and integration suites during deployments.
Developer Tooling & Workflows
Accelerate developer onboarding with pre-configured project utilities.
Open Source Extension
Fork and customize internal modules under the repository's open Apache-2.0 license.
🚀 Getting Started
Install optax using your package manager: `pip install optax`
Initialize your project workspace or configuration file for optax.
Import optax into your codebase or invoke it directly from your terminal.
Execute your test suite or run `optax --help` to verify successful setup.
👍 Strengths
⚠️ Considerations
⇄ Alternatives & Direct Competitors
👥 Who Should Use This
Developers and engineering teams building with Python, seeking reliable, tested, and actively maintained tooling for production workloads.
🏆 Nearby in the Rankings
google-deepmind/optax is currently ranked #2,426 by stars across every repository tracked on GitHubRepo. These are adjacent projects:
| Rank | Repository | Language | Stars | Action |
|---|---|---|---|---|
| #2,420 | david942j/one_gadget | Ruby | ★ 2.4K | Compare ↗ |
| #2,422 | velopack/velopack | C# | ★ 2.4K | Compare ↗ |
| #2,423 | Taiizor/ReaLTaiizor | C# | ★ 2.3K | Compare ↗ |
| #2,423 | jest-community/jest-extended | TypeScript | ★ 2.3K | Compare ↗ |
| #2,425 | googleprojectzero/fuzzilli | Swift | ★ 2.3K | Compare ↗ |
| #2,426 | google-deepmind/optax This Project | Python | ★ 2.3K | |
| #2,427 | ncdai/chanhdai.com | TypeScript | ★ 2.3K | Compare ↗ |
| #2,428 | Xed-Editor/Xed-Editor | Kotlin | ★ 2.3K | Compare ↗ |
| #2,428 | thoas/picfit | Go | ★ 2.3K | Compare ↗ |
| #2,430 | eugr/spark-vllm-docker | Python | ★ 2.3K | Compare ↗ |
| #2,430 | pelican/panel | PHP | ★ 2.3K | Compare ↗ |
Frequently Asked Questions
What does optax do? +
Optax is a gradient processing and optimization library for JAX.
What language is optax written in? +
The primary language is Python. Topics include: machine-learning, optimization.
Is optax actively maintained? +
Yes, the last recorded push was on Oct 1, 2026 with 129 open issues being tracked.
How many stars does optax have? +
optax has 2,345 stars and 372 forks on GitHub.
How does optax rank among GitHub repositories? +
With 2,345 stars, google-deepmind/optax is ranked #2,426 globally across all repositories tracked on GitHubRepo and #350 among Python projects.
What license is optax distributed under? +
The repository reports a Apache-2.0 license. Always verify the repository LICENSE file for legal terms.
