noahgolmant / pytorch-hessian-eigenthingsLinks
Efficient PyTorch Hessian eigendecomposition tools!
☆379Updated last year
Alternatives and similar repositories for pytorch-hessian-eigenthings
Users that are interested in pytorch-hessian-eigenthings are comparing it to the libraries listed below
Sorting:
- ☆157Updated 3 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆277Updated 2 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆215Updated 3 weeks ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆147Updated 2 years ago
- hessian in pytorch☆187Updated 4 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.