verbiiyo / rigl-torchLinks
Lightweight torch implementation of rigl, a sparse-to-sparse optimizer.
☆57Updated 3 years ago
Alternatives and similar repositories for rigl-torch
Users that are interested in rigl-torch are comparing it to the libraries listed below
Sorting:
- [ICML2022] Training Your Sparse Neural Network Better with Any Mask. Ajay Jaiswal, Haoyu Ma, Tianlong Chen, ying Ding, and Zhangyang Wang☆28Updated 2 years ago
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆105Updated 5 years ago
- ☆226Updated 11 months ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆52Updated 4 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆91Updated 2 years ago
- PyTorch implementation of Mixer-nano (#parameters is 0.67M, originally Mixer-S/16 has 18M) with 90.83 % acc. on CIFAR-10. Training from s…☆33Updated 3 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆108Updated 6 years ago
- Code for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot☆42Updated 4 years ago
- ☆77Updated last year
- ☆14Updated 4 years ago
- [IJCAI'22 Survey] Recent Advances on Neural Network Pruning at Initialization.☆59Updated last year
- Code release for REPAIR: REnormalizing Permuted Activations for Interpolation Repair☆48Updated last year
- ☆191Updated 4 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Updated 2 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 6 years ago
- Introducing diverse tasks for NAS☆50Updated 2 years ago
- ☆209Updated 2 years ago
- Distributed K-FAC preconditioner for PyTorch☆87Updated last week
- Implementation of Continuous Sparsification, a method for pruning and ticket search in deep networks☆33Updated 3 years ago
- [ICML 2021] "Do We Actually Need Dense Over-Parameterization? In-Time Over-Parameterization in Sparse Training" by Shiwei Liu, Lu Yin, De…☆45Updated last year
- HW-GPT-Bench: Hardware-Aware Architecture Benchmark for Language Models☆20Updated 7 months ago
- ☆67Updated 6 years ago
- Reproducing RigL (ICML 2020) as a part of ML Reproducibility Challenge 2020☆28Updated 3 years ago
- ☆28Updated last year
- Efficient Riemannian Optimization on Stiefel Manifold via Cayley Transform☆41Updated 6 years ago
- ☆70Updated 7 months ago
- [Neurips 2021] Sparse Training via Boosting Pruning Plasticity with Neuroregeneration☆31Updated 2 years ago
- ☆83Updated 5 years ago
- Implementation of Effective Sparsification of Neural Networks with Global Sparsity Constraint☆31Updated 3 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆215Updated 3 weeks ago