.conda_env.yml
.gitignore
.readthedocs.yml
LICENSE.txt
README-dev.md
README.md
black.toml
changelog.md
fully_documented.txt
makefile
pytest.ini
setup.cfg
setup.py
.github/workflows/lint.yaml
.github/workflows/test.yaml
backpack/__init__.py
backpack/context.py
backpack/core/__init__.py
backpack/core/derivatives/__init__.py
backpack/core/derivatives/avgpool1d.py
backpack/core/derivatives/avgpool2d.py
backpack/core/derivatives/avgpool3d.py
backpack/core/derivatives/avgpoolnd.py
backpack/core/derivatives/basederivatives.py
backpack/core/derivatives/batchnorm1d.py
backpack/core/derivatives/conv1d.py
backpack/core/derivatives/conv2d.py
backpack/core/derivatives/conv3d.py
backpack/core/derivatives/conv_transpose1d.py
backpack/core/derivatives/conv_transpose2d.py
backpack/core/derivatives/conv_transpose3d.py
backpack/core/derivatives/conv_transposend.py
backpack/core/derivatives/convnd.py
backpack/core/derivatives/crossentropyloss.py
backpack/core/derivatives/dropout.py
backpack/core/derivatives/elementwise.py
backpack/core/derivatives/elu.py
backpack/core/derivatives/flatten.py
backpack/core/derivatives/leakyrelu.py
backpack/core/derivatives/linear.py
backpack/core/derivatives/logsigmoid.py
backpack/core/derivatives/maxpool1d.py
backpack/core/derivatives/maxpool2d.py
backpack/core/derivatives/maxpool3d.py
backpack/core/derivatives/maxpoolnd.py
backpack/core/derivatives/mseloss.py
backpack/core/derivatives/relu.py
backpack/core/derivatives/selu.py
backpack/core/derivatives/shape_check.py
backpack/core/derivatives/sigmoid.py
backpack/core/derivatives/tanh.py
backpack/core/derivatives/zeropad2d.py
backpack/extensions/__init__.py
backpack/extensions/backprop_extension.py
backpack/extensions/curvature.py
backpack/extensions/mat_to_mat_jac_base.py
backpack/extensions/module_extension.py
backpack/extensions/curvmatprod/__init__.py
backpack/extensions/curvmatprod/ggnmp/__init__.py
backpack/extensions/curvmatprod/ggnmp/activations.py
backpack/extensions/curvmatprod/ggnmp/batchnorm1d.py
backpack/extensions/curvmatprod/ggnmp/conv2d.py
backpack/extensions/curvmatprod/ggnmp/dropout.py
backpack/extensions/curvmatprod/ggnmp/flatten.py
backpack/extensions/curvmatprod/ggnmp/ggnmpbase.py
backpack/extensions/curvmatprod/ggnmp/linear.py
backpack/extensions/curvmatprod/ggnmp/losses.py
backpack/extensions/curvmatprod/ggnmp/padding.py
backpack/extensions/curvmatprod/ggnmp/pooling.py
backpack/extensions/curvmatprod/hmp/__init__.py
backpack/extensions/curvmatprod/hmp/activations.py
backpack/extensions/curvmatprod/hmp/batchnorm1d.py
backpack/extensions/curvmatprod/hmp/conv2d.py
backpack/extensions/curvmatprod/hmp/dropout.py
backpack/extensions/curvmatprod/hmp/flatten.py
backpack/extensions/curvmatprod/hmp/hmpbase.py
backpack/extensions/curvmatprod/hmp/linear.py
backpack/extensions/curvmatprod/hmp/losses.py
backpack/extensions/curvmatprod/hmp/padding.py
backpack/extensions/curvmatprod/hmp/pooling.py
backpack/extensions/curvmatprod/pchmp/__init__.py
backpack/extensions/curvmatprod/pchmp/activations.py
backpack/extensions/curvmatprod/pchmp/conv2d.py
backpack/extensions/curvmatprod/pchmp/dropout.py
backpack/extensions/curvmatprod/pchmp/flatten.py
backpack/extensions/curvmatprod/pchmp/linear.py
backpack/extensions/curvmatprod/pchmp/losses.py
backpack/extensions/curvmatprod/pchmp/padding.py
backpack/extensions/curvmatprod/pchmp/pchmpbase.py
backpack/extensions/curvmatprod/pchmp/pooling.py
backpack/extensions/firstorder/__init__.py
backpack/extensions/firstorder/base.py
backpack/extensions/firstorder/batch_grad/__init__.py
backpack/extensions/firstorder/batch_grad/batch_grad_base.py
backpack/extensions/firstorder/batch_grad/batchnorm1d.py
backpack/extensions/firstorder/batch_grad/conv1d.py
backpack/extensions/firstorder/batch_grad/conv2d.py
backpack/extensions/firstorder/batch_grad/conv3d.py
backpack/extensions/firstorder/batch_grad/conv_transpose1d.py
backpack/extensions/firstorder/batch_grad/conv_transpose2d.py
backpack/extensions/firstorder/batch_grad/conv_transpose3d.py
backpack/extensions/firstorder/batch_grad/linear.py
backpack/extensions/firstorder/batch_l2_grad/__init__.py
backpack/extensions/firstorder/batch_l2_grad/conv1d.py
backpack/extensions/firstorder/batch_l2_grad/conv2d.py
backpack/extensions/firstorder/batch_l2_grad/conv3d.py
backpack/extensions/firstorder/batch_l2_grad/convnd.py
backpack/extensions/firstorder/batch_l2_grad/convtranspose1d.py
backpack/extensions/firstorder/batch_l2_grad/convtranspose2d.py
backpack/extensions/firstorder/batch_l2_grad/convtranspose3d.py
backpack/extensions/firstorder/batch_l2_grad/convtransposend.py
backpack/extensions/firstorder/batch_l2_grad/linear.py
backpack/extensions/firstorder/gradient/__init__.py
backpack/extensions/firstorder/gradient/base.py
backpack/extensions/firstorder/gradient/batchnorm1d.py
backpack/extensions/firstorder/gradient/conv1d.py
backpack/extensions/firstorder/gradient/conv2d.py
backpack/extensions/firstorder/gradient/conv3d.py
backpack/extensions/firstorder/gradient/convtranspose1d.py
backpack/extensions/firstorder/gradient/convtranspose2d.py
backpack/extensions/firstorder/gradient/convtranspose3d.py
backpack/extensions/firstorder/gradient/linear.py
backpack/extensions/firstorder/sum_grad_squared/__init__.py
backpack/extensions/firstorder/sum_grad_squared/conv1d.py
backpack/extensions/firstorder/sum_grad_squared/conv2d.py
backpack/extensions/firstorder/sum_grad_squared/conv3d.py
backpack/extensions/firstorder/sum_grad_squared/convtranspose1d.py
backpack/extensions/firstorder/sum_grad_squared/convtranspose2d.py
backpack/extensions/firstorder/sum_grad_squared/convtranspose3d.py
backpack/extensions/firstorder/sum_grad_squared/linear.py
backpack/extensions/firstorder/sum_grad_squared/sgs_base.py
backpack/extensions/firstorder/variance/__init__.py
backpack/extensions/firstorder/variance/conv1d.py
backpack/extensions/firstorder/variance/conv2d.py
backpack/extensions/firstorder/variance/conv3d.py
backpack/extensions/firstorder/variance/convtranspose1d.py
backpack/extensions/firstorder/variance/convtranspose2d.py
backpack/extensions/firstorder/variance/convtranspose3d.py
backpack/extensions/firstorder/variance/linear.py
backpack/extensions/firstorder/variance/variance_base.py
backpack/extensions/secondorder/__init__.py
backpack/extensions/secondorder/diag_ggn/__init__.py
backpack/extensions/secondorder/diag_ggn/activations.py
backpack/extensions/secondorder/diag_ggn/conv1d.py
backpack/extensions/secondorder/diag_ggn/conv2d.py
backpack/extensions/secondorder/diag_ggn/conv3d.py
backpack/extensions/secondorder/diag_ggn/convnd.py
backpack/extensions/secondorder/diag_ggn/convtranspose1d.py
backpack/extensions/secondorder/diag_ggn/convtranspose2d.py
backpack/extensions/secondorder/diag_ggn/convtranspose3d.py
backpack/extensions/secondorder/diag_ggn/convtransposend.py
backpack/extensions/secondorder/diag_ggn/diag_ggn_base.py
backpack/extensions/secondorder/diag_ggn/dropout.py
backpack/extensions/secondorder/diag_ggn/flatten.py
backpack/extensions/secondorder/diag_ggn/linear.py
backpack/extensions/secondorder/diag_ggn/losses.py
backpack/extensions/secondorder/diag_ggn/padding.py
backpack/extensions/secondorder/diag_ggn/pooling.py
backpack/extensions/secondorder/diag_hessian/__init__.py
backpack/extensions/secondorder/diag_hessian/activations.py
backpack/extensions/secondorder/diag_hessian/conv1d.py
backpack/extensions/secondorder/diag_hessian/conv2d.py
backpack/extensions/secondorder/diag_hessian/conv3d.py
backpack/extensions/secondorder/diag_hessian/convnd.py
backpack/extensions/secondorder/diag_hessian/convtranspose1d.py
backpack/extensions/secondorder/diag_hessian/convtranspose2d.py
backpack/extensions/secondorder/diag_hessian/convtranspose3d.py
backpack/extensions/secondorder/diag_hessian/convtransposend.py
backpack/extensions/secondorder/diag_hessian/diag_h_base.py
backpack/extensions/secondorder/diag_hessian/dropout.py
backpack/extensions/secondorder/diag_hessian/flatten.py
backpack/extensions/secondorder/diag_hessian/linear.py
backpack/extensions/secondorder/diag_hessian/losses.py
backpack/extensions/secondorder/diag_hessian/padding.py
backpack/extensions/secondorder/diag_hessian/pooling.py
backpack/extensions/secondorder/hbp/__init__.py
backpack/extensions/secondorder/hbp/activations.py
backpack/extensions/secondorder/hbp/conv2d.py
backpack/extensions/secondorder/hbp/dropout.py
backpack/extensions/secondorder/hbp/flatten.py
backpack/extensions/secondorder/hbp/hbp_options.py
backpack/extensions/secondorder/hbp/hbpbase.py
backpack/extensions/secondorder/hbp/linear.py
backpack/extensions/secondorder/hbp/losses.py
backpack/extensions/secondorder/hbp/padding.py
backpack/extensions/secondorder/hbp/pooling.py
backpack/hessianfree/__init__.py
backpack/hessianfree/ggnvp.py
backpack/hessianfree/hvp.py
backpack/hessianfree/lop.py
backpack/hessianfree/rop.py
backpack/utils/__init__.py
backpack/utils/conv.py
backpack/utils/conv_transpose.py
backpack/utils/convert_parameters.py
backpack/utils/examples.py
backpack/utils/hooks.py
backpack/utils/kroneckers.py
backpack/utils/linear.py
backpack/utils/unsqueeze.py
backpack_for_pytorch.egg-info/PKG-INFO
backpack_for_pytorch.egg-info/SOURCES.txt
backpack_for_pytorch.egg-info/dependency_links.txt
backpack_for_pytorch.egg-info/not-zip-safe
backpack_for_pytorch.egg-info/requires.txt
backpack_for_pytorch.egg-info/top_level.txt
docs/.nojekyll
docs/CNAME
docs/examples.html
docs/index.html
docs/jekyll-theme-minimal.gemspec
docs/assets/dangel2020backpack.bib
docs/assets/css/style.css
docs/assets/fonts/Noto-Sans-700/Noto-Sans-700.eot
docs/assets/fonts/Noto-Sans-700/Noto-Sans-700.svg
docs/assets/fonts/Noto-Sans-700/Noto-Sans-700.ttf
docs/assets/fonts/Noto-Sans-700/Noto-Sans-700.woff
docs/assets/fonts/Noto-Sans-700/Noto-Sans-700.woff2
docs/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.eot
docs/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.svg
docs/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.ttf
docs/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.woff
docs/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.woff2
docs/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.eot
docs/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.svg
docs/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.ttf
docs/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.woff
docs/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.woff2
docs/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.eot
docs/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.svg
docs/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.ttf
docs/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.woff
docs/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.woff2
docs/assets/img/backpack_logo_torch.svg
docs/assets/img/logo.png
docs/assets/img/updaterule.png
docs/assets/js/scale.fix.js
docs/script/bootstrap
docs/script/cibuild
docs/script/release
docs/script/validate-html
docs_src/.gitignore
docs_src/README.md
docs_src/buildweb.sh
docs_src/examples/cheatsheet.pdf
docs_src/examples/basic_usage/README.rst
docs_src/examples/basic_usage/example_all_in_one.py
docs_src/examples/use_cases/README.rst
docs_src/examples/use_cases/example_cg_newton.py
docs_src/examples/use_cases/example_custom_module.py
docs_src/examples/use_cases/example_diag_ggn_optimizer.py
docs_src/examples/use_cases/example_differential_privacy.py
docs_src/examples/use_cases/example_extension_hook.py
docs_src/examples/use_cases/example_first_order_resnet.py
docs_src/examples/use_cases/example_gradient_of_variance.py
docs_src/examples/use_cases/example_save_memory_convolutions.py
docs_src/examples/use_cases/example_trace_estimation.py
docs_src/rtd/.gitignore
docs_src/rtd/.nojekyll
docs_src/rtd/Makefile
docs_src/rtd/conf.py
docs_src/rtd/extensions.rst
docs_src/rtd/good-to-know.rst
docs_src/rtd/index.rst
docs_src/rtd/main-api.rst
docs_src/rtd/make.bat
docs_src/rtd/supported-layers.rst
docs_src/rtd/torch.inventory
docs_src/rtd/assets/backpack_logo_torch.png
docs_src/splash/.gitignore
docs_src/splash/Gemfile
docs_src/splash/_config.yml
docs_src/splash/examples.md
docs_src/splash/index.md
docs_src/splash/jekyll-theme-minimal.gemspec
docs_src/splash/_includes/code-samples.html
docs_src/splash/_includes/dangel2020backpack.bib
docs_src/splash/_layouts/default.html
docs_src/splash/_layouts/post.html
docs_src/splash/_sass/fonts.scss
docs_src/splash/_sass/jekyll-theme-minimal.scss
docs_src/splash/_sass/rouge-github.scss
docs_src/splash/assets/dangel2020backpack.bib
docs_src/splash/assets/css/style.css
docs_src/splash/assets/fonts/Noto-Sans-700/Noto-Sans-700.eot
docs_src/splash/assets/fonts/Noto-Sans-700/Noto-Sans-700.svg
docs_src/splash/assets/fonts/Noto-Sans-700/Noto-Sans-700.ttf
docs_src/splash/assets/fonts/Noto-Sans-700/Noto-Sans-700.woff
docs_src/splash/assets/fonts/Noto-Sans-700/Noto-Sans-700.woff2
docs_src/splash/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.eot
docs_src/splash/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.svg
docs_src/splash/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.ttf
docs_src/splash/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.woff
docs_src/splash/assets/fonts/Noto-Sans-700italic/Noto-Sans-700italic.woff2
docs_src/splash/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.eot
docs_src/splash/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.svg
docs_src/splash/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.ttf
docs_src/splash/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.woff
docs_src/splash/assets/fonts/Noto-Sans-italic/Noto-Sans-italic.woff2
docs_src/splash/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.eot
docs_src/splash/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.svg
docs_src/splash/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.ttf
docs_src/splash/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.woff
docs_src/splash/assets/fonts/Noto-Sans-regular/Noto-Sans-regular.woff2
docs_src/splash/assets/img/backpack_logo_torch.svg
docs_src/splash/assets/img/logo.png
docs_src/splash/assets/img/updaterule.png
docs_src/splash/assets/js/scale.fix.js
docs_src/splash/script/bootstrap
docs_src/splash/script/cibuild
docs_src/splash/script/release
docs_src/splash/script/validate-html
logo/backpack_logo_no_torch.svg
logo/backpack_logo_torch.svg
test/__init__.py
test/automated_kfac_test.py
test/automated_test.py
test/bugfixes_test.py
test/conv2d_test.py
test/interface_test.py
test/layers.py
test/layers_test.py
test/linear_test.py
test/networks.py
test/problems.py
test/readme.md
test/test___init__.py
test/test_problem.py
test/test_problems_activations.py
test/test_problems_bn.py
test/test_problems_convolutions.py
test/test_problems_kfacs.py
test/test_problems_linear.py
test/test_problems_padding.py
test/test_problems_pooling.py
test/test_second_order_warnings.py
test/test_simple_resnet.py
test/utils_test.py
test/benchmark/__init__.py
test/benchmark/functionality.py
test/benchmark/jvp.py
test/benchmark/jvp_activations.py
test/benchmark/jvp_avgpool2d.py
test/benchmark/jvp_conv2d.py
test/benchmark/jvp_linear.py
test/benchmark/jvp_maxpool2d.py
test/benchmark/jvp_zeropad2d.py
test/core/__init__.py
test/core/derivatives/__init__.py
test/core/derivatives/activation_settings.py
test/core/derivatives/convolution_settings.py
test/core/derivatives/derivatives_test.py
test/core/derivatives/linear_settings.py
test/core/derivatives/loss_settings.py
test/core/derivatives/padding_settings.py
test/core/derivatives/pooling_settings.py
test/core/derivatives/problem.py
test/core/derivatives/settings.py
test/core/derivatives/utils.py
test/core/derivatives/implementation/autograd.py
test/core/derivatives/implementation/backpack.py
test/core/derivatives/implementation/base.py
test/extensions/__init__.py
test/extensions/automated_settings.py
test/extensions/problem.py
test/extensions/test_backprop_extension.py
test/extensions/test_hooks.py
test/extensions/firstorder/__init__.py
test/extensions/firstorder/firstorder_settings.py
test/extensions/firstorder/batch_grad/__init__.py
test/extensions/firstorder/batch_grad/batchgrad_settings.py
test/extensions/firstorder/batch_grad/test_batchgrad.py
test/extensions/firstorder/batch_l2_grad/__init__.py
test/extensions/firstorder/batch_l2_grad/batchl2grad_settings.py
test/extensions/firstorder/batch_l2_grad/test_batchl2grad.py
test/extensions/firstorder/sum_grad_squared/__init__.py
test/extensions/firstorder/sum_grad_squared/sumgradsquared_settings.py
test/extensions/firstorder/sum_grad_squared/test_sumgradsquared.py
test/extensions/firstorder/variance/__init__.py
test/extensions/firstorder/variance/test_variance.py
test/extensions/firstorder/variance/variance_settings.py
test/extensions/implementation/__init__.py
test/extensions/implementation/autograd.py
test/extensions/implementation/backpack.py
test/extensions/implementation/base.py
test/extensions/implementation/hooks.py
test/extensions/secondorder/__init__.py
test/extensions/secondorder/secondorder_settings.py
test/extensions/secondorder/diag_ggn/__init__.py
test/extensions/secondorder/diag_ggn/diaggnn_settings.py
test/extensions/secondorder/diag_ggn/test_batch_diag_ggn.py
test/extensions/secondorder/diag_ggn/test_diag_ggn.py
test/extensions/secondorder/diag_hessian/__init__.py
test/extensions/secondorder/diag_hessian/diagh_settings.py
test/extensions/secondorder/diag_hessian/test_diag_hessian.py
test/extensions/secondorder/hbp/__init__.py
test/extensions/secondorder/hbp/kfac_settings.py
test/extensions/secondorder/hbp/kflr_settings.py
test/extensions/secondorder/hbp/kfra_settings.py
test/extensions/secondorder/hbp/test_kfac.py
test/extensions/secondorder/hbp/test_kflr.py
test/extensions/secondorder/hbp/test_kfra.py
test/implementation/__init__.py
test/implementation/implementation.py
test/implementation/implementation_autograd.py
test/implementation/implementation_bpext.py
test/utils/__init__.py
test/utils/test_conv.py
test/utils/test_conv_settings.py
test/utils/test_conv_transpose.py
test/utils/test_conv_transpose_settings.py