yandex-research / moshpit-sgdLinks
"Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation
☆29Updated 7 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:
- "Towards Crowdsourced Training of Large Neural Networks using Decentralized Mixture-of-Experts" (NeurIPS 2020), original PyTorch implemen…☆57Updated 4 years ago
- A Learnable LSH Framework for Efficient NN Training☆32Updated 4 years ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆16Updated 3 years ago
- Factorized Neural Layers☆30Updated 2 years ago
- Hyperparameter tuning via uncertainty modeling☆47Updated last year
- ☆29Updated 2 years ago
- A simple implementation of a deep linear Pytorch module☆21Updated 4 years ago
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆47Updated last year
- [NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks☆60Updated 2 years ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆68Updated 9 months ago
- Code associated with the paper **Fine-tuning Language Models over Slow Networks using Activation Compression with Guarantees**.☆28Updated 2 years ago
- Distributed K-FAC preconditioner for PyTorch☆90Updated this week
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 4 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆50Updated 3 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆46Updated 5 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 4 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- NeurIPS 2021 - Few-shot learning competition☆26Updated 3 years ago
- Latest Weight Averaging (NeurIPS HITY 2022)☆31Updated 2 years ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- ☆27Updated 2 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆51Updated last year
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 2 years ago
- Blog post☆17Updated last year
- ModelDiff: A Framework for Comparing Learning Algorithms☆59Updated 2 years ago
- ☆37Updated 3 years ago
- ☆29Updated 2 years ago