optax

PyPI page
Home page
Author: None
Summary: A gradient processing and optimization library in JAX.
Latest version: 0.2.4
Required dependencies: absl-py | chex | etils | jax | jaxlib | numpy
Optional dependencies: absl-py | attrs | dm-tree | dp_accounting | flax | ipython | ipywidgets | matplotlib | mpmath | myst-nb | numpy | scikit-learn | scipy | sphinx | sphinx-autodoc-typehints | sphinx-book-theme | sphinx-collections | sphinx-gallery | sphinx_contributors | sphinxcontrib-katex | tensorflow | tensorflow-datasets

Downloads last day: 35,080
Downloads last week: 255,204
Downloads last month: 1,121,430