epfml / powersgd
Practical low-rank gradient compression for distributed optimization: https://arxiv.org/abs/1905.13727
☆146Updated 4 months ago
Alternatives and similar repositories for powersgd:
Users that are interested in powersgd are comparing it to the libraries listed below
- Implementation of (overlap) local SGD in Pytorch☆33Updated 4 years ago
- GRACE - GRAdient ComprEssion for distributed deep learning☆138Updated 8 months ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 3 years ago
- ☆46Updated 5 years ago
- [ICLR 2018] Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Training☆217Updated 8 months ago
- Atomo: Communication-efficient Learning via Atomic Sparsification☆25Updated 6 years ago
- ☆74Updated 5 years ago
- Research and development for optimizing transformers☆125Updated 4 years ago
- Sketched SGD☆28Updated 4 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- QSGD-TF☆21Updated 5 years ago
- Stochastic Gradient Push for Distributed Deep Learning☆160Updated last year
- gTop-k S-SGD: A Communication-Efficient Distributed Synchronous SGD Algorithm for Deep Learning☆36Updated 5 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- Sparsified SGD with Memory: https://arxiv.org/abs/1809.07599☆59Updated 6 years ago
- ☆93Updated 2 years ago
- Decentralized SGD and Consensus with Communication Compression: https://arxiv.org/abs/1907.09356☆66Updated 4 years ago
- Code for the signSGD paper☆83Updated 4 years ago
- ☆202Updated 2 years ago
- FTPipe and related pipeline model parallelism research.☆41Updated last year
- implement distributed machine learning with Pytorch + OpenMPI☆51Updated 6 years ago
- Efficient reference implementations of the static & dynamic M-FAC algorithms (for pruning and optimization)☆16Updated 3 years ago
- ☆101Updated 7 months ago
- ☆72Updated 3 years ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆38Updated 2 years ago
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆74Updated 4 years ago
- Simple Distributed Deep Learning on TensorFlow☆134Updated 2 years ago
- ☆26Updated 3 years ago
- Accuracy 77%. Large batch deep learning optimizer LARS for ImageNet with PyTorch and ResNet, using Horovod for distribution. Optional acc…☆38Updated 3 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago