PyPI page
Author:
DeepMind
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTI...
Summary:
A Jax package for approximate curvature estimation and optimization using KFAC.
Latest version:
0.0.8
Required dependencies:
absl-py
|
distrax
|
dm-tree
|
immutabledict
|
jax
|
jaxlib
|
numpy
|
optax
|
typing-extensions
Optional dependencies:
absl-py
|
distrax
|
dm-haiku
|
dm-tree
|
docutils
|
immutabledict
|
ipykernel
|
ipython
|
jax
|
jaxlib
|
jaxline
|
matplotlib
|
ml-collections
|
myst-nb
|
numpy
|
optax
|
pandoc
|
pytest-xdist
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-rtd-theme
|
sphinxcontrib-bibtex
|
sphinxcontrib-katex
|
tensorflow
|
tensorflow-datasets
Downloads last day:
59
Downloads last week:
3,321
Downloads last month:
6,623
7-day smoothing