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:
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Updated 2 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆282Updated 3 years ago
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆105Updated 5 years ago
- ☆158Updated 3 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆383Updated last year
- hessian in pytorch☆187Updated 5 years ago
- Hessian spectral density estimation in TF and Jax☆124Updated 5 years ago
- ☆194Updated last month
- an implementation of L0 regularization with PyTorch☆57Updated 7 years ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆282Updated 2 years ago
- ☆145Updated 2 years ago
- ☆227Updated last year
- Labels and other data for the paper "Are we done with ImageNet?"☆199Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- ☆126Updated last year
- A Re-implementation of Fixed-update Initialization☆155Updated 6 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Updated 5 years ago
- ☆182Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
- Implementation for the Lookahead Optimizer.☆243Updated 3 years ago
- Optimization with orthogonal constraints and on general manifolds☆130Updated 5 years ago
- ☆28Updated 3 years ago
- Demonstration and tutorial notebooks for the Higra library☆13Updated 3 months ago
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆129Updated 2 years ago
- Implementation of the reversible residual network in pytorch☆106Updated 4 years ago
- Hypergradient descent☆147Updated last year
- ☆83Updated 5 years ago
- Release of CIFAR-10.1, a new test set for CIFAR-10.☆225Updated 5 years ago
- PyTorch Framework for Developing Memory Efficient Deep Invertible Networks☆256Updated 2 years ago
- Pytorch implementation of Variational Dropout Sparsifies Deep Neural Networks☆84Updated 4 years ago