noahgolmant / pytorch-hessian-eigenthings
Efficient PyTorch Hessian eigendecomposition tools!
☆355Updated 6 months ago
Related projects: ⓘ
- ☆153Updated 2 years ago
- hessian in pytorch☆185Updated 3 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆262Updated last year
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆555Updated 4 months ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆139Updated last year
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆204Updated last month
- Hessian spectral density estimation in TF and Jax☆112Updated 4 years ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆415Updated 2 years ago
- Code for experiments in my blog post on the Neural Tangent Kernel: https://rajatvd.github.io/NTK☆166Updated 4 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆125Updated 5 years ago
- NTK reading group☆84Updated 4 years ago
- ☆118Updated 3 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆141Updated 11 months ago
- Release of CIFAR-10.1, a new test set for CIFAR-10.☆218Updated 4 years ago
- Totally Versatile Miscellanea for Pytorch☆461Updated 2 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆674Updated 5 months ago
- code for "Residual Flows for Invertible Generative Modeling".☆260Updated last year
- Laplace approximations for Deep Learning.☆447Updated this week
- Hypergradient descent☆137Updated 3 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆176Updated last month
- Convolutional Neural Tangent Kernel☆108Updated 4 years ago
- Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch☆293Updated 9 months ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆135Updated 5 years ago
- ☆157Updated last month
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆263Updated last year
- PyTorch implementations of algorithms for density estimation☆570Updated 3 years ago
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆124Updated last year
- ☆182Updated 3 years ago
- ☆231Updated last year
- Code for experiments regarding importance sampling for training neural networks☆320Updated 2 years ago