LICENSE
MANIFEST.in
README.md
requirements.txt
requirements_docs.txt
requirements_examples.txt
requirements_tests.txt
setup.py
docs/conf.py
docs/ext/coverage_check.py
examples/autoencoder_mnist/experiment.py
examples/autoencoder_mnist/pipeline.py
examples/classifier_mnist/experiment.py
examples/classifier_mnist/pipeline.py
examples/lrelunet101_imagenet/experiment.py
examples/lrelunet101_imagenet/pipeline.py
examples/resnet50_imagenet/experiment.py
examples/resnet50_imagenet/pipeline.py
kfac_jax/__init__.py
kfac_jax/py.typed
kfac_jax.egg-info/PKG-INFO
kfac_jax.egg-info/SOURCES.txt
kfac_jax.egg-info/dependency_links.txt
kfac_jax.egg-info/not-zip-safe
kfac_jax.egg-info/requires.txt
kfac_jax.egg-info/top_level.txt
kfac_jax/_src/__init__.py
kfac_jax/_src/curvature_blocks.py
kfac_jax/_src/curvature_estimator.py
kfac_jax/_src/layers_and_loss_tags.py
kfac_jax/_src/loss_functions.py
kfac_jax/_src/optimizer.py
kfac_jax/_src/patches_second_moment.py
kfac_jax/_src/tag_graph_matcher.py
kfac_jax/_src/tracer.py
kfac_jax/_src/utils.py