epfml / LocalSGD-Code
☆42Updated 4 years ago
Related projects: ⓘ
- Implementation of (overlap) local SGD in Pytorch☆32Updated 4 years ago
- Code for the signSGD paper☆79Updated 3 years ago
- Sparsified SGD with Memory: https://arxiv.org/abs/1809.07599☆54Updated 5 years ago
- ☆72Updated 5 years ago
- SGD with compressed gradients and error-feedback: https://arxiv.org/abs/1901.09847☆29Updated last month
- Atomo: Communication-efficient Learning via Atomic Sparsification☆25Updated 5 years ago
- vector quantization for stochastic gradient descent.☆33Updated 4 years ago
- FedNAS: Federated Deep Learning via Neural Architecture Search☆50Updated 3 years ago
- Partial implementation of paper "DEEP GRADIENT COMPRESSION: REDUCING THE COMMUNICATION BANDWIDTH FOR DISTRIBUTED TRAINING"☆31Updated 3 years ago
- Sketched SGD☆28Updated 4 years ago
- Code for "Adaptive Gradient Quantization for Data-Parallel SGD", published in NeurIPS 2020.☆28Updated 3 years ago
- Decentralized SGD and Consensus with Communication Compression: https://arxiv.org/abs/1907.09356☆63Updated 4 years ago
- GRACE - GRAdient ComprEssion for distributed deep learning☆134Updated last month
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆97Updated 4 years ago
- Practical low-rank gradient compression for distributed optimization: https://arxiv.org/abs/1905.13727☆140Updated 2 weeks ago
- Code and checkpoints of compressed networks for the paper titled "HYDRA: Pruning Adversarially Robust Neural Networks" (NeurIPS 2020) (ht…☆88Updated last year
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆88Updated last year
- A compressed adaptive optimizer for training large-scale deep learning models using PyTorch☆27Updated 4 years ago
- Code for paper: Variance Reduced Local SGD with Lower Communication Complexity☆12Updated 4 years ago
- Federated posterior averaging implemented in JAX☆49Updated last year
- Simple Hierarchical Count Sketch in Python☆20Updated 3 years ago
- Stochastic Gradient Push for Distributed Deep Learning☆156Updated last year
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆101Updated 5 years ago
- QSGD-TF☆21Updated 5 years ago
- Pytorch deep learning object detection using CINIC-10 dataset.☆22Updated 4 years ago
- Efficient reference implementations of the static & dynamic M-FAC algorithms (for pruning and optimization)☆16Updated 2 years ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆54Updated 3 years ago
- SNIP: SINGLE-SHOT NETWORK PRUNING BASED ON CONNECTION SENSITIVITY☆108Updated 5 years ago
- implement distributed machine learning with Pytorch + OpenMPI☆52Updated 5 years ago
- [ICLR 2018] Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Training☆209Updated 2 months ago