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/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/maxpool2d.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/convtranspose1d.py
backpack/extensions/firstorder/batch_l2_grad/convtranspose2d.py
backpack/extensions/firstorder/batch_l2_grad/convtranspose3d.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/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/conv_transpose.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_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_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/extensions/__init__.py
test/extensions/problem.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/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