DachengLi1 / AMP
(NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
☆38Updated 2 years ago
Alternatives and similar repositories for AMP:
Users that are interested in AMP are comparing it to the libraries listed below
- ☆72Updated 3 years ago
- nnScaler: Compiling DNN models for Parallel Training☆97Updated 2 weeks ago
- ☆75Updated 2 years ago
- Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning☆23Updated 2 months ago
- ☆50Updated 8 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 3 months ago
- Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.☆59Updated last year
- FTPipe and related pipeline model parallelism research.☆41Updated last year
- A resilient distributed training framework☆88Updated 10 months ago
- ☆79Updated 2 years ago
- [ICML 2024] Serving LLMs on heterogeneous decentralized clusters.☆21Updated 9 months ago
- DISB is a new DNN inference serving benchmark with diverse workloads and models, as well as real-world traces.☆53Updated 6 months ago
- TileFusion is a highly efficient kernel template library designed to elevate the level of abstraction in CUDA C for processing tiles.☆60Updated this week
- Official repository for the paper DynaPipe: Optimizing Multi-task Training through Dynamic Pipelines☆17Updated last year
- ☆71Updated 3 months ago
- Stateful LLM Serving☆46Updated 7 months ago
- ☆83Updated 3 months ago
- Python package for rematerialization-aware gradient checkpointing☆24Updated last year
- ☆16Updated 2 years ago
- [IJCAI2023] An automated parallel training system that combines the advantages from both data and model parallelism. If you have any inte…☆51Updated last year
- An experimental parallel training platform☆54Updated 11 months ago
- ☆24Updated last year
- FlexFlow Serve: Low-Latency, High-Performance LLM Serving☆22Updated this week
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆74Updated 4 years ago
- AlpaServe: Statistical Multiplexing with Model Parallelism for Deep Learning Serving (OSDI 23)☆82Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆106Updated 3 months ago
- ☆47Updated 2 months ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 3 years ago
- ☆14Updated 10 months ago
- A Cluster-Wide Model Manager to Accelerate DNN Training via Automated Training Warmup☆34Updated 2 years ago