hwang595 / ps_pytorchLinks
implement distributed machine learning with Pytorch + OpenMPI
☆51Updated 6 years ago
Alternatives and similar repositories for ps_pytorch
Users that are interested in ps_pytorch are comparing it to the libraries listed below
Sorting:
- Ternary Gradients to Reduce Communication in Distributed Deep Learning (TensorFlow)☆182Updated 6 years ago
- ☆75Updated 6 years ago
- Sparsified SGD with Memory: https://arxiv.org/abs/1809.07599☆58Updated 6 years ago
- Stochastic Gradient Push for Distributed Deep Learning☆170Updated 2 years ago
- Implementation of Parameter Server using PyTorch communication lib☆42Updated 6 years ago
- Implementing Google's DistBelief paper☆110Updated 2 years ago
- A compressed adaptive optimizer for training large-scale deep learning models using PyTorch☆27Updated 5 years ago
- ☆53Updated 7 years ago
- QSGD-TF☆21Updated 6 years ago
- ☆12Updated 6 years ago
- Official code for "Writing Distributed Applications with PyTorch", PyTorch Tutorial☆263Updated 2 years ago
- Path-Level Network Transformation for Efficient Architecture Search, in ICML 2018.☆112Updated 7 years ago
- An analytical performance modeling tool for deep neural networks.☆89Updated 4 years ago
- PyTorch parameter server with MPI☆16Updated 7 years ago
- Efficient Architecture Search by Network Transformation, in AAAI 2018☆169Updated 6 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 5 years ago
- papers on scalable and efficient machine learning systems☆192Updated 6 years ago
- Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"☆144Updated 8 years ago
- GPU-specialized parameter server for GPU machine learning.☆101Updated 7 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- A Tool for Automatic Parallelization of Deep Learning Training in Distributed Multi-GPU Environments.☆132Updated 3 years ago
- Structured Bayesian Pruning, NIPS 2017☆74Updated 5 years ago
- Tutorial code on how to build your own Deep Learning System in 2k Lines☆125Updated 8 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 6 years ago
- Code for ICML 2017 paper, SplitNet: Learning to Semantically Split Deep Networks for Parameter Reduction and Model Parallelization☆54Updated 7 years ago
- Sketched SGD☆28Updated 5 years ago
- Implementation of ICLR 2017 paper "Loss-aware Binarization of Deep Networks"☆18Updated 6 years ago
- Atomo: Communication-efficient Learning via Atomic Sparsification☆27Updated 6 years ago
- CS294; AI For Systems and Systems For AI☆224Updated 5 years ago
- Practical low-rank gradient compression for distributed optimization: https://arxiv.org/abs/1905.13727☆147Updated 8 months ago