LeviViana / torchessianLinks
Full loss Hessian spectrum approximation tool.
☆13Updated 6 years ago
Alternatives and similar repositories for torchessian
Users that are interested in torchessian are comparing it to the libraries listed below
Sorting:
- ☆193Updated this week
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Updated 2 years ago
- Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)☆16Updated 6 years ago
- ☆157Updated 3 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆149Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
- Hessian spectral density estimation in TF and Jax☆124Updated 5 years ago
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆105Updated 5 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆382Updated last year
- Percentile computation for pytorch☆21Updated 5 years ago
- ☆227Updated last year
- Prescribed Generative Adversarial Networks☆143Updated 5 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆168Updated 3 years ago
- Hypergradient descent☆147Updated last year
- Labels and other data for the paper "Are we done with ImageNet?"☆198Updated 4 years ago
- ☆133Updated 4 years ago
- A Re-implementation of Fixed-update Initialization☆155Updated 6 years ago
- CIFAR-5m dataset☆39Updated 4 years ago
- ☆28Updated 3 years ago
- ☆100Updated 4 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆281Updated 3 years ago
- an implementation of L0 regularization with PyTorch☆57Updated 7 years ago
- ☆144Updated 2 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 6 years ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆281Updated 2 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆149Updated 2 years ago
- Implementation of the reversible residual network in pytorch☆106Updated 3 years ago
- ☆83Updated 5 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆53Updated 4 years ago