DachengLi1 / AMPView external linksLinks
(NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
☆44Nov 4, 2022Updated 3 years ago
Alternatives and similar repositories for AMP
Users that are interested in AMP are comparing it to the libraries listed below
Sorting:
- DELTA-pytorch：DELTA: Dynamically Optimizing GPU Memory beyond Tensor Recomputation☆12Apr 16, 2024Updated last year
- An experimental parallel training platform☆56Mar 25, 2024Updated last year
- ☆77May 4, 2021Updated 4 years ago
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆76Dec 11, 2020Updated 5 years ago
- ☆12Apr 30, 2024Updated last year
- ☆26Dec 5, 2022Updated 3 years ago
- Yaksa: High-performance Noncontiguous Data Management☆14Oct 1, 2025Updated 4 months ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆147Jun 25, 2022Updated 3 years ago
- Binary Neural Network-based COVID-19 Face-Mask Wear and Positioning Predictor on Edge Devices☆12Jul 1, 2021Updated 4 years ago
- Bamboo is a system for running large pipeline-parallel DNNs affordably, reliably, and efficiently using spot instances.☆55Dec 11, 2022Updated 3 years ago
- Hi-Speed DNN Training with Espresso: Unleashing the Full Potential of Gradient Compression with Near-Optimal Usage Strategies (EuroSys '2…☆15Sep 21, 2023Updated 2 years ago
- ☆82Updated this week
- ☆17Dec 8, 2023Updated 2 years ago
- Benchmarks☆17Apr 28, 2025Updated 9 months ago
- FTPipe and related pipeline model parallelism research.☆44May 16, 2023Updated 2 years ago
- GPTQ inference TVM kernel☆40Apr 25, 2024Updated last year
- ☆22Apr 22, 2024Updated last year
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆20Feb 23, 2024Updated last year
- ☆17Oct 17, 2025Updated 3 months ago
- ☆13Feb 22, 2023Updated 2 years ago
- A PIM instrumentation, compilation, execution, simulation, and evaluation repository for BLIMP-style architectures.☆18May 12, 2022Updated 3 years ago
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆17Mar 13, 2023Updated 2 years ago
- nnScaler: Compiling DNN models for Parallel Training☆124Sep 23, 2025Updated 4 months ago
- High performance NCCL plugin for Bagua.☆15Sep 15, 2021Updated 4 years ago
- Python package for rematerialization-aware gradient checkpointing☆27Oct 31, 2023Updated 2 years ago
- ☆84Dec 2, 2022Updated 3 years ago
- ☆22Dec 15, 2023Updated 2 years ago
- Torch Frontend for IREE☆25Dec 21, 2023Updated 2 years ago
- A schedule language for large model training☆152Aug 21, 2025Updated 5 months ago
- A resilient distributed training framework☆96Apr 11, 2024Updated last year
- TACOS: [T]opology-[A]ware [Co]llective Algorithm [S]ynthesizer for Distributed Machine Learning☆31Jun 13, 2025Updated 8 months ago
- Github repo backing website for the CS Assistant Professor Handbook☆27Aug 28, 2024Updated last year
- Distributed ML Optimizer☆35Jul 28, 2021Updated 4 years ago
- Create and deploy virtual-experiments - co-processing computational workflows☆10Jan 28, 2026Updated 2 weeks ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆82Nov 19, 2024Updated last year
- ☆71Mar 26, 2025Updated 10 months ago
- ☆36Dec 13, 2023Updated 2 years ago
- An awesome curated list of languages and tools to program FPGAs☆73Jun 22, 2022Updated 3 years ago
- A PyTorch implementation of the paper "Decoupled Parallel Backpropagation with Convergence Guarantee"☆30Jul 18, 2018Updated 7 years ago