yandex-research / moshpit-sgdLinks
"Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation
☆29Updated 6 months ago
Alternatives and similar repositories for moshpit-sgd
Users that are interested in moshpit-sgd are comparing it to the libraries listed below
Sorting:
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 4 years ago
- A Learnable LSH Framework for Efficient NN Training☆32Updated 4 years ago
- Code associated with the paper **Fine-tuning Language Models over Slow Networks using Activation Compression with Guarantees**.☆28Updated 2 years ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆16Updated 3 years ago
- "Towards Crowdsourced Training of Large Neural Networks using Decentralized Mixture-of-Experts" (NeurIPS 2020), original PyTorch implemen…☆57Updated 4 years ago
- Deadline-based hyperparameter tuning on RayTune.☆31Updated 5 years ago
- Distributed K-FAC preconditioner for PyTorch☆89Updated this week
- Memory-efficient transformer. Work in progress.☆19Updated 2 years ago
- ☆22Updated 4 years ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- A Sparse-tensor Communication Framework for Distributed Deep Learning☆13Updated 3 years ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- ☆75Updated 2 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago
- ☆14Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 4 years ago
- ☆20Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 4 years ago
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆46Updated 5 years ago
- Torch Distributed Experimental☆117Updated last year
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 2 years ago
- Dynamic Neural Architecture Search Toolkit☆30Updated 8 months ago
- Factorized Neural Layers☆29Updated 2 years ago
- ☆14Updated 4 years ago
- Hacks for PyTorch☆19Updated 2 years ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 4 years ago