5 projects
backpack-for-pytorch
BackPACK: Packing more into backprop
unfoldNd
N-dimensional unfold (im2col) and fold (col2im) in PyTorch
curvlinops-for-pytorch
scipy Linear operator implementations of the GGN and Hessian in PyTorch
vivit-for-pytorch
ViViT: Curvature access through the generalized Gauss-Newton\'s low-rank structure
cockpit-for-pytorch
A Practical Debugging Tool for Training Deep Neural Networks.