PyPI page
Author:
Google DeepMind
Summary:
A gradient processing and optimization library in JAX.
Latest version:
0.2.8
Required dependencies:
absl-py
|
jax
|
jaxlib
|
numpy
Optional dependencies:
flax
|
ipython
|
matplotlib
|
myst-nb
|
scikit-learn
|
scipy
|
setuptools
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-book-theme
|
sphinx-collections
|
sphinx-contributors
|
sphinx-gallery
|
sphinxcontrib-katex
Downloads last day:
133,930
Downloads last week:
979,577
Downloads last month:
2,907,174
7-day smoothing