Distributed K-FAC preconditioner for PyTorch
☆98Oct 27, 2026Updated this week
Alternatives and similar repositories for kfac-pytorch
Users that are interested in kfac-pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- BERT for Distributed PyTorch + AMP Training☆12Mar 15, 2023Updated 3 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆135Jul 2, 2019Updated 7 years ago
- A Chainer extension for K-FAC☆20Jun 16, 2019Updated 7 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆154Jun 22, 2023Updated 3 years ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆285Feb 27, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- An implementation of KFAC for TensorFlow☆198Feb 11, 2022Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆150Oct 1, 2023Updated 3 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆225Oct 13, 2026Updated 2 weeks ago
- Regularization, Neural Network Training Dynamics☆14Jan 13, 2020Updated 6 years ago
- SKFAC Preconditioner for MindSpore☆12Jul 2, 2021Updated 5 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆795Jul 10, 2025Updated last year
- [ICLR 2023] Eva: Practical Second-order Optimization with Kronecker-vectorized Approximation☆12Jul 31, 2023Updated 3 years ago
- ☆33Jul 8, 2024Updated 2 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆193Dec 5, 2024Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆14Sep 14, 2021Updated 5 years ago
- Code to simulate energy-based analog systems and equilibrium propagation☆37Apr 6, 2025Updated last year
- ☆32Feb 11, 2021Updated 5 years ago
- Efficient reference implementations of the static & dynamic M-FAC algorithms (for pruning and optimization)☆17Feb 23, 2022Updated 4 years ago
- Pytorch optimizers implementing Hilbert Constrained Gradient Descent☆19May 9, 2019Updated 7 years ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆24Nov 4, 2024Updated last year
- ☆25Mar 1, 2022Updated 4 years ago
- Ok-Topk is a scheme for distributed training with sparse gradients. Ok-Topk integrates a novel sparse allreduce algorithm (less than 6k c…☆27Dec 10, 2022Updated 3 years ago
- ☆11Apr 29, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Group project "Algorithms for large-scale optimal transport". Implement ADMMs and Sinkhorn's Algorithms.☆11Jan 28, 2019Updated 7 years ago
- Artifact for IPDPS'21: DSXplore: Optimizing Convolutional Neural Networks via Sliding-Channel Convolutions.☆13Apr 6, 2021Updated 5 years ago
- Accuracy 77%. Large batch deep learning optimizer LARS for ImageNet with PyTorch and ResNet, using Horovod for distribution. Optional acc…☆38Jun 1, 2021Updated 5 years ago
- A decentralised application that creates high quality machine learning datasets☆12Jan 22, 2019Updated 7 years ago
- Natural Gradient, Variational Inference☆29Jan 13, 2020Updated 6 years ago
- Randomized algorithm class at CU☆18Jul 8, 2025Updated last year
- [ICLR 2025] AdaFisher: Adaptive Second Order Optimization via Fisher Information☆54Feb 7, 2025Updated last year
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆1,000Jan 30, 2024Updated 2 years ago
- EP☆21Mar 9, 2021Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Anonymized code for Igeood: An Information Geometry Approach to Out-of-Distribution Detection☆12Jan 25, 2022Updated 4 years ago
- Repository to reproduce the results of the paper "Holomorphic Equilibrium Propagation Computes Exact Gradients Through Finite Size Oscill…☆11Oct 20, 2024Updated 2 years ago
- ☆46Nov 18, 2022Updated 3 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆15Nov 4, 2024Updated last year
- PyTorch implementation of LAMB for ImageNet/ResNet-50 training☆13May 13, 2021Updated 5 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆54Mar 8, 2021Updated 5 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆182Jul 17, 2021Updated 5 years ago