YiwenShaoStephen / NGD-SGD
A Pytorch Implementation of Natural Gradient Descent
☆42Updated 5 years ago
Related projects ⓘ
Alternatives and complementary repositories for NGD-SGD
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆140Updated last year
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆127Updated 5 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆135Updated 5 years ago
- ☆59Updated 3 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆69Updated 3 months ago
- Create animations for the optimization trajectory of neural nets☆137Updated 9 months ago
- Limitations of the Empirical Fisher Approximation☆45Updated 4 years ago
- Distributed K-FAC Preconditioner for PyTorch☆80Updated this week
- PyTorch AutoNEB implementation to identify minimum energy paths, e.g. in neural network loss landscapes☆54Updated 2 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆206Updated last month
- Discrete Normalizing Flows implemented in PyTorch☆107Updated 3 years ago
- Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK☆166Updated 5 years ago
- NTK reading group☆86Updated 4 years ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Updated 6 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Optimization with orthogonal constraints and on general manifolds☆126Updated 4 years ago
- Pytorch library for factorized L0-based pruning.☆43Updated last year
- ☆119Updated 5 months ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆97Updated 4 years ago
- ☆57Updated last year
- Hessian spectral density estimation in TF and Jax☆115Updated 4 years ago
- pyhessian is a TensorFlow module which can be used to estimate Hessian matrices☆23Updated 3 years ago
- Speech2Vec Reality Check☆75Updated last year
- Pytorch implementation of the Power Spherical distribution☆73Updated 3 months ago
- Convex potential flows☆78Updated 2 years ago
- Representation learning for NLP @ JSALT19☆36Updated 4 years ago
- A Pytorch implementation of an efficient unitary neural network (https://arxiv.org/abs/1612.05231)☆32Updated 4 years ago
- Structured matrices for compressing neural networks☆67Updated last year
- Code for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot☆43Updated 4 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆35Updated 2 years ago