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:
42,223
Downloads last week:
257,870
Downloads last month:
1,030,068