nelaturuharsha / TurboPrune
Harness for training/finding lottery tickets in PyTorch. With support for multiple pruning techniques and augmented by distributed training, FFCV and AMP.
☆16Updated last month
Alternatives and similar repositories for TurboPrune:
Users that are interested in TurboPrune are comparing it to the libraries listed below
- Fair Generative Modeling via Weak Supervision☆20Updated 2 years ago
- Python library for argument and configuration management☆53Updated 2 years ago
- Sketched matrix decompositions for PyTorch☆70Updated last month
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆32Updated this week
- Learning to See by Looking at Noise☆108Updated 3 months ago
- ☆42Updated last year
- Train ImageNet *fast* in 500 lines of code with FFCV☆139Updated 9 months ago
- ☆16Updated 2 years ago
- Implements stochastic line search☆118Updated last year
- Code for Spectral Norm of Convolutional Layers with Circular and Zero Paddings and Efficient Bound of Lipschitz Constant for Convolutiona…☆14Updated last year
- Create animations for the optimization trajectory of neural nets☆146Updated last year
- SGD with large step sizes learns sparse features [ICML 2023]☆32Updated last year
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆125Updated last year
- Slides from previous talks.☆29Updated last year
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆268Updated 2 years ago
- Binary classification algorithm under fairness considerations.☆9Updated 2 years ago
- Pytorch implementation of COIN++ 🍁☆69Updated 2 years ago
- ☆17Updated 8 months ago
- CIFAR-5m dataset☆38Updated 4 years ago
- Implementation of "Structured Multi-Hashing for Model Compression" (CVPR 2020)☆11Updated 4 years ago
- ☆58Updated last year
- PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)☆52Updated 2 years ago
- Differentially-private Wasserstein GAN implementation in PyTorch☆27Updated 5 years ago
- code to reproduce the empirical results in the research paper☆35Updated 3 years ago
- ☆121Updated 9 months ago
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- The code for the paper: https://arxiv.org/pdf/1802.00168.pdf☆17Updated 5 years ago
- The M2L school 2022 tutorials☆36Updated 2 years ago
- Official code for the Stochastic Polyak step-size optimizer☆138Updated 8 months ago
- Algorithms for Privacy-Preserving Machine Learning in JAX☆92Updated 8 months ago