.conda_env.yml
.coveragerc
.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/python-publish.yml
.github/workflows/test.yaml
backpack/__init__.py
backpack/context.py
backpack/core/__init__.py
backpack/core/derivatives/__init__.py
backpack/core/derivatives/adaptive_avg_pool_nd.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/batchnorm_nd.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/embedding.py
backpack/core/derivatives/flatten.py
backpack/core/derivatives/leakyrelu.py
backpack/core/derivatives/linear.py
backpack/core/derivatives/logsigmoid.py
backpack/core/derivatives/lstm.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/pad.py
backpack/core/derivatives/permute.py
backpack/core/derivatives/relu.py
backpack/core/derivatives/rnn.py
backpack/core/derivatives/scale_module.py
backpack/core/derivatives/selu.py
backpack/core/derivatives/shape_check.py
backpack/core/derivatives/sigmoid.py
backpack/core/derivatives/slicing.py
backpack/core/derivatives/sum_module.py
backpack/core/derivatives/tanh.py
backpack/core/derivatives/zeropad2d.py
backpack/custom_module/__init__.py
backpack/custom_module/branching.py
backpack/custom_module/graph_utils.py
backpack/custom_module/pad.py
backpack/custom_module/permute.py
backpack/custom_module/reduce_tuple.py
backpack/custom_module/scale_module.py
backpack/custom_module/slicing.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/saved_quantities.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/batchnorm_nd.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/embedding.py
backpack/extensions/firstorder/batch_grad/linear.py
backpack/extensions/firstorder/batch_grad/rnn.py
backpack/extensions/firstorder/batch_l2_grad/__init__.py
backpack/extensions/firstorder/batch_l2_grad/batch_l2_base.py
backpack/extensions/firstorder/batch_l2_grad/batchnorm_nd.py
backpack/extensions/firstorder/batch_l2_grad/convnd.py
backpack/extensions/firstorder/batch_l2_grad/convtransposend.py
backpack/extensions/firstorder/batch_l2_grad/embedding.py
backpack/extensions/firstorder/batch_l2_grad/linear.py
backpack/extensions/firstorder/batch_l2_grad/rnn.py
backpack/extensions/firstorder/gradient/__init__.py
backpack/extensions/firstorder/gradient/base.py
backpack/extensions/firstorder/gradient/batchnorm_nd.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/embedding.py
backpack/extensions/firstorder/gradient/linear.py
backpack/extensions/firstorder/gradient/rnn.py
backpack/extensions/firstorder/sum_grad_squared/__init__.py
backpack/extensions/firstorder/sum_grad_squared/batchnorm_nd.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/embedding.py
backpack/extensions/firstorder/sum_grad_squared/linear.py
backpack/extensions/firstorder/sum_grad_squared/rnn.py
backpack/extensions/firstorder/sum_grad_squared/sgs_base.py
backpack/extensions/firstorder/variance/__init__.py
backpack/extensions/firstorder/variance/batchnorm_nd.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/embedding.py
backpack/extensions/firstorder/variance/linear.py
backpack/extensions/firstorder/variance/rnn.py
backpack/extensions/firstorder/variance/variance_base.py
backpack/extensions/secondorder/__init__.py
backpack/extensions/secondorder/base.py
backpack/extensions/secondorder/diag_ggn/__init__.py
backpack/extensions/secondorder/diag_ggn/activations.py
backpack/extensions/secondorder/diag_ggn/adaptive_avg_pool_nd.py
backpack/extensions/secondorder/diag_ggn/batchnorm_nd.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/custom_module.py
backpack/extensions/secondorder/diag_ggn/diag_ggn_base.py
backpack/extensions/secondorder/diag_ggn/dropout.py
backpack/extensions/secondorder/diag_ggn/embedding.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/pad.py
backpack/extensions/secondorder/diag_ggn/padding.py
backpack/extensions/secondorder/diag_ggn/permute.py
backpack/extensions/secondorder/diag_ggn/pooling.py
backpack/extensions/secondorder/diag_ggn/rnn.py
backpack/extensions/secondorder/diag_ggn/slicing.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/pad.py
backpack/extensions/secondorder/diag_hessian/padding.py
backpack/extensions/secondorder/diag_hessian/pooling.py
backpack/extensions/secondorder/diag_hessian/slicing.py
backpack/extensions/secondorder/hbp/__init__.py
backpack/extensions/secondorder/hbp/activations.py
backpack/extensions/secondorder/hbp/conv2d.py
backpack/extensions/secondorder/hbp/custom_module.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/extensions/secondorder/sqrt_ggn/__init__.py
backpack/extensions/secondorder/sqrt_ggn/activations.py
backpack/extensions/secondorder/sqrt_ggn/base.py
backpack/extensions/secondorder/sqrt_ggn/batchnorm_nd.py
backpack/extensions/secondorder/sqrt_ggn/convnd.py
backpack/extensions/secondorder/sqrt_ggn/convtransposend.py
backpack/extensions/secondorder/sqrt_ggn/custom_module.py
backpack/extensions/secondorder/sqrt_ggn/dropout.py
backpack/extensions/secondorder/sqrt_ggn/embedding.py
backpack/extensions/secondorder/sqrt_ggn/flatten.py
backpack/extensions/secondorder/sqrt_ggn/linear.py
backpack/extensions/secondorder/sqrt_ggn/losses.py
backpack/extensions/secondorder/sqrt_ggn/pad.py
backpack/extensions/secondorder/sqrt_ggn/padding.py
backpack/extensions/secondorder/sqrt_ggn/pooling.py
backpack/extensions/secondorder/sqrt_ggn/slicing.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/errors.py
backpack/utils/examples.py
backpack/utils/hooks.py
backpack/utils/kroneckers.py
backpack/utils/linear.py
backpack/utils/module_classification.py
backpack/utils/subsampling.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_resnet_all_in_one.py
docs_src/examples/use_cases/example_rnn.py
docs_src/examples/use_cases/example_save_memory_convolutions.py
docs_src/examples/use_cases/example_subsampling.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/rtd/assets/backpack_logo_torch.svg
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_batch_first.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_retain_graph.py
test/test_second_order_warnings.py
test/utils_test.py
test/adaptive_avg_pool/__init__.py
test/adaptive_avg_pool/problem.py
test/adaptive_avg_pool/settings_adaptive_avg_pool_nd.py
test/adaptive_avg_pool/test_adaptive_avg_pool_nd.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/converter/__init__.py
test/converter/converter_cases.py
test/converter/resnet_cases.py
test/converter/test_converter.py
test/core/__init__.py
test/core/derivatives/__init__.py
test/core/derivatives/activation_settings.py
test/core/derivatives/batch_norm_settings.py
test/core/derivatives/convolution_settings.py
test/core/derivatives/derivatives_test.py
test/core/derivatives/embedding_settings.py
test/core/derivatives/linear_settings.py
test/core/derivatives/loss_settings.py
test/core/derivatives/lstm_settings.py
test/core/derivatives/padding_settings.py
test/core/derivatives/permute_settings.py
test/core/derivatives/pooling_adaptive_settings.py
test/core/derivatives/pooling_settings.py
test/core/derivatives/problem.py
test/core/derivatives/rnn_settings.py
test/core/derivatives/scale_module_settings.py
test/core/derivatives/settings.py
test/core/derivatives/slicing_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/custom_module/__init__.py
test/custom_module/test_pad.py
test/custom_module/test_slicing.py
test/extensions/__init__.py
test/extensions/automated_settings.py
test/extensions/graph_clear_test.py
test/extensions/problem.py
test/extensions/test_backprop_extension.py
test/extensions/test_hooks.py
test/extensions/utils.py
test/extensions/firstorder/__init__.py
test/extensions/firstorder/firstorder_settings.py
test/extensions/firstorder/batch_grad/__init__.py
test/extensions/firstorder/batch_grad/batch_grad_settings.py
test/extensions/firstorder/batch_grad/test_batch_grad.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/diag_ggn_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/extensions/secondorder/sqrt_ggn/__init__.py
test/extensions/secondorder/sqrt_ggn/sqrt_ggn_settings.py
test/extensions/secondorder/sqrt_ggn/test_sqrt_ggn.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/evaluation_mode.py
test/utils/skip_test.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
test/utils/test_subsampling.py