Package profile
optax
- Summary: A gradient processing and optimization library in JAX.
- Author: Google DeepMind <optax-dev@google.com>
- Homepage: https://github.com/google-deepmind/optax
- Source: https://github.com/google-deepmind/optax (Repo profile)
- Number of releases: 23
- First release: 0.0.1 on 2020-08-14
- Latest release: 0.2.5 on 2025-06-10
Releases
PyPI Downloads
Dependencies
Optax has 18 dependencies, 13 of which optional.| Dependency | Optional |
|---|---|
| absl-py | false |
| chex | false |
| jax | false |
| jaxlib | false |
| numpy | false |
| flax | true |
| ipython | true |
| matplotlib | true |
| myst-nb | true |
| scikit-learn | true |
| scipy | true |
| sphinx | true |
| sphinx_contributors | true |
| sphinx-autodoc-typehints | true |
| sphinx-book-theme | true |
| sphinx-collections | true |
| sphinx-gallery | true |
| sphinxcontrib-katex | true |