README.md
setup.py
backpack/__init__.py
backpack/context.py
backpack/core/__init__.py
backpack/core/derivatives/__init__.py
backpack/core/derivatives/avgpool2d.py
backpack/core/derivatives/basederivatives.py
backpack/core/derivatives/batchnorm1d.py
backpack/core/derivatives/conv2d.py
backpack/core/derivatives/crossentropyloss.py
backpack/core/derivatives/dropout.py
backpack/core/derivatives/elementwise.py
backpack/core/derivatives/flatten.py
backpack/core/derivatives/linear.py
backpack/core/derivatives/maxpool2d.py
backpack/core/derivatives/mseloss.py
backpack/core/derivatives/relu.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/activations.py
backpack/extensions/curvmatprod/batchnorm1d.py
backpack/extensions/curvmatprod/cmpbase.py
backpack/extensions/curvmatprod/conv2d.py
backpack/extensions/curvmatprod/dropout.py
backpack/extensions/curvmatprod/flatten.py
backpack/extensions/curvmatprod/linear.py
backpack/extensions/curvmatprod/losses.py
backpack/extensions/curvmatprod/padding.py
backpack/extensions/curvmatprod/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/conv2d.py
backpack/extensions/firstorder/batch_grad/linear.py
backpack/extensions/firstorder/batch_l2_grad/__init__.py
backpack/extensions/firstorder/batch_l2_grad/conv2d.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/conv2d.py
backpack/extensions/firstorder/gradient/linear.py
backpack/extensions/firstorder/sum_grad_squared/__init__.py
backpack/extensions/firstorder/sum_grad_squared/conv2d.py
backpack/extensions/firstorder/sum_grad_squared/linear.py
backpack/extensions/firstorder/variance/__init__.py
backpack/extensions/firstorder/variance/conv2d.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/conv2d.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/conv2d.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/convert_parameters.py
backpack/utils/ein.py
backpack/utils/examples.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
test/__init__.py
test/automated_bn_test.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/test_ea_jac_t_mat_jac_prod.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/test_sqrt_hessian.py
test/test_sqrt_hessian_sampled.py
test/test_sum_hessian.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/implementation/__init__.py
test/implementation/implementation.py
test/implementation/implementation_autograd.py
test/implementation/implementation_bpext.py