Package search
optax
- Summary: A gradient processing and optimization library in JAX.
- Author:
- Homepage: https://github.com/google-deepmind/optax
- Source: https://github.com/google-deepmind/optax
- Repo profile
- Number of releases: 22
- First release: 0.0.1 on 2020-08-14T12:42:30
- Latest release: 0.2.4 on 2024-11-12T21:52:15
| Dependency | Optional |
|---|---|
| absl-py | false |
| chex | false |
| etils | false |
| jax | false |
| jaxlib | false |
| numpy | false |
| attrs | true |
| dm-tree | true |
| dp_accounting | true |
| flax | true |
| ipython | true |
| ipywidgets | true |
| matplotlib | true |
| mpmath | 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 |
| tensorflow | true |
| tensorflow-datasets | true |