Tabrizian / learning-to-quantize
Code for "Adaptive Gradient Quantization for Data-Parallel SGD", published in NeurIPS 2020.
☆30Updated 4 years ago
Alternatives and similar repositories for learning-to-quantize:
Users that are interested in learning-to-quantize are comparing it to the libraries listed below
- Partial implementation of paper "DEEP GRADIENT COMPRESSION: REDUCING THE COMMUNICATION BANDWIDTH FOR DISTRIBUTED TRAINING"☆31Updated 4 years ago
- vector quantization for stochastic gradient descent.☆34Updated 4 years ago
- ☆46Updated 5 years ago
- Code for the signSGD paper☆84Updated 4 years ago
- SGD with compressed gradients and error-feedback: https://arxiv.org/abs/1901.09847☆30Updated 9 months ago
- Sparsified SGD with Memory: https://arxiv.org/abs/1809.07599☆58Updated 6 years ago
- Implementation of (overlap) local SGD in Pytorch☆33Updated 4 years ago
- FedNAS: Federated Deep Learning via Neural Architecture Search☆54Updated 3 years ago
- Create tiny ML systems for on-device learning.☆20Updated 3 years ago
- Understanding Top-k Sparsification in Distributed Deep Learning☆24Updated 5 years ago
- ☆29Updated 5 years ago
- Federated Dynamic Sparse Training☆30Updated 2 years ago
- [ICLR 2018] Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Training☆221Updated 9 months ago
- Atomo: Communication-efficient Learning via Atomic Sparsification☆25Updated 6 years ago
- ☆74Updated 5 years ago
- Recycling Model Updates in Federated Learning: Are Gradient Subspaces Low-Rank?☆14Updated 3 years ago
- Model compression by constrained optimization, using the Learning-Compression (LC) algorithm☆73Updated 3 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆107Updated 6 years ago
- [Neurips 2021] Sparse Training via Boosting Pruning Plasticity with Neuroregeneration☆31Updated 2 years ago
- LotteryFL: Empower Edge Intelligence with Personalized and Communication-Efficient Federated Learning (2021 IEEE/ACM Symposium on Edge Co…☆42Updated 2 years ago
- Code and checkpoints of compressed networks for the paper titled "HYDRA: Pruning Adversarially Robust Neural Networks" (NeurIPS 2020) (ht…☆92Updated 2 years ago
- Implementation of Compressed SGD with Compressed Gradients in Pytorch☆12Updated 9 months ago
- It is implementation of Research paper "DEEP GRADIENT COMPRESSION: REDUCING THE COMMUNICATION BANDWIDTH FOR DISTRIBUTED TRAINING". Deep g…☆18Updated 5 years ago
- Codes for paper "Few Shot Network Compression via Cross Distillation", AAAI 2020.☆31Updated 5 years ago
- Any-Precision Deep Neural Networks (AAAI 2021)☆60Updated 5 years ago
- Measuring and predicting on-device metrics (latency, power, etc.) of machine learning models☆66Updated 2 years ago
- ☆128Updated last year
- [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
- gTop-k S-SGD: A Communication-Efficient Distributed Synchronous SGD Algorithm for Deep Learning☆36Updated 5 years ago
- ☆23Updated 4 years ago