google-deepmind / optax
Optax is a gradient processing and optimization library for JAX.
Why this job: works in the areas we cover, but does not say which job it does
What it does
Optax is a gradient processing and optimization library for JAX.
Get started
pip install optax- Maintenance
- Active
- Last commit today
- Stars
- 2,366
- +4 this week
- Licence
- Apache-2.0
- Allows business use
- Downloads for
- None listed
- No release files
- Latest version
- v0.2.9
- Released yesterday
- Built with
- Python
- 100% of the code
Is it looked after?
- Contributors
- 13
- Last 90 days
- Issues
- 91% closed
- 34 still open
- Pull requests
- 804 merged
- 134 waiting
- Releases
- 25
- 3 in the last year
- Contributing guide (opens in new tab)
- No security policy
- No code of conduct
Similar repos
| Repo | Stars | Last commit | Licence | Downloads for |
|---|---|---|---|---|
| mlpack / ensmallen | 814 | 5 days ago | Custom | No downloads |
| PrunaAI / pruna | 1,386 | 14 days ago | Apache-2.0 | No downloads |
| emdgroup / baybe | 528 | yesterday | Apache-2.0 | No downloads |
More: languages, topics, project record and preview image
Languages
- Python 100%
- Shell <1%
Topics
- machine-learning
- optimization
Project record
- JobOther AI tools
- StatusListed
- Commits2,070
- Forks395
- Watching32
- Stars in 28 days+19
- CreatedJun 12, 2020
- First seen by usNov 7, 2026
- CheckedNov 12, 2026