PyPI Stats

Search

All packages
Top packages

Track packages

optax


PyPI page
Home page
Author: DeepMind
License: Apache 2.0
Summary: A gradient processing and optimisation library in JAX.
Latest version: 0.0.9
Required dependencies: absl-py | chex | jax | jaxlib | numpy

Downloads last day: 195,801
Downloads last week: 500,736
Downloads last month: 915,412