amirgholami / adahessianView external linksLinks
ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
☆283Feb 27, 2023Updated 2 years ago
Alternatives and similar repositories for adahessian
Users that are interested in adahessian are comparing it to the libraries listed below
Sorting:
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Nov 12, 2020Updated 5 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆775Jul 10, 2025Updated 7 months ago
- Distributed K-FAC preconditioner for PyTorch☆95Updated this week
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆218Apr 4, 2021Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Oct 1, 2023Updated 2 years ago
- Jax implementation of the AdaHessian optimizer☆20Mar 11, 2021Updated 4 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,161Mar 22, 2024Updated last year
- ☆23Mar 1, 2022Updated 3 years ago
- diffGrad: An Optimization Method for Convolutional Neural Networks☆55Oct 12, 2022Updated 3 years ago
- ☆43Jan 30, 2024Updated 2 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆604Nov 28, 2025Updated 2 months ago
- Scalable Computation of Hessian Diagonals☆14Jun 2, 2024Updated last year
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆221Updated this week
- Limitations of the Empirical Fisher Approximation☆49Mar 3, 2025Updated 11 months ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆53Mar 8, 2021Updated 4 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆384Feb 29, 2024Updated last year
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- Ranger deep learning optimizer rewrite to use newest components☆340Feb 18, 2024Updated last year
- Analyze AdaHessian optimizer on 2D functions.☆13Aug 13, 2021Updated 4 years ago
- An implementation of KFAC for TensorFlow☆199Feb 11, 2022Updated 4 years ago
- (unofficial) - customized fork of DETR, optimized for intelligent obj detection on 'real world' custom datasets☆12Aug 22, 2020Updated 5 years ago
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆981Jan 30, 2024Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Aug 6, 2021Updated 4 years ago
- An adaptive training algorithm for residual network☆17Aug 22, 2020Updated 5 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,961Feb 21, 2024Updated last year
- Apollo: An Adaptive Parameter-wise Diagonal Quasi-Newton Method for Nonconvex Stochastic Optimization☆182Nov 21, 2021Updated 4 years ago
- Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"☆1,068Aug 9, 2024Updated last year
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Jul 2, 2019Updated 6 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- [ICML2023] Instant Soup Cheap Pruning Ensembles in A Single Pass Can Draw Lottery Tickets from Large Models. Ajay Jaiswal, Shiwei Liu, Ti…☆11Nov 28, 2023Updated 2 years ago
- [ICML 2021] The official PyTorch Implementations of Positive-Negative Momentum Optimizers.☆27Aug 30, 2022Updated 3 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Jul 17, 2021Updated 4 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆152Jun 22, 2023Updated 2 years ago
- Manifold-Mixup implementation for fastai V1☆19Oct 1, 2020Updated 5 years ago
- Hypergradient descent☆147May 31, 2024Updated last year
- ☆83Jan 15, 2020Updated 6 years ago
- A Chainer extension for K-FAC☆20Jun 16, 2019Updated 6 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆15Nov 4, 2024Updated last year