optax
Optax is a gradient processing and optimization library for JAX.
Build custom optimization tools for machine learning using simple components that you can mix and match.
Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.
Get an email alert on its next release or when it starts trending — never miss the moment.
Free · no card · unsubscribe anytimeOptax is a gradient processing and optimization library for JAX.
optax has 2.3k stars on GitHub. It has been forked 346 times. optax is written mainly in Python. It has been in active development since 2020. optax is available under the Apache-2.0 license. Its main topics are machine-learning, optimization.
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.
What license does optax use?
optax is available under the Apache-2.0 license.
What language is optax written in?
optax is written mainly in Python.
Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.
[](https://olud.ai/project/google-deepmind-optax.html)
Measured from GitHub topics shared by both projects, weighted by how rare each topic is.