PyPI Stats

Search

All packages
Top packages

Track packages

optax


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

Downloads last day: 269,981
Downloads last week: 1,886,075
Downloads last month: 8,623,931