PipeDream: Fast and Efficient Pipeline Parallel DNN Training
Not on the map. Why?
Abstract (arXiv)
PipeDream is a Deep Neural Network(DNN) training system for GPUs that parallelizes computation by pipelining execution across multiple machines. Its pipeline parallel computing model avoids the slowdowns faced by data-parallel training when large models and/or limited network bandwidth induce high communication-to-computation ratios. PipeDream reduces communication by up to 95% for large DNNs relative to data-parallel training, and allows perfect overlap of communication and computation. PipeDream keeps all available GPUs productive by systematically partitioning DNN layers among them to balance work and minimize communication, versions model parameters for backward pass correctness, and schedules the forward and backward passes of different inputs in round-robin fashion to optimize "time to target accuracy". Experiments with five different DNNs on two different clusters show that PipeDream is up to 5x faster in time-to-accuracy compared to data-parallel training.
Key facts. Cited by 116 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Jun 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 20; 2022: 11; 2023: 20; 2024: 24; 2025: 28; 2026: 12.
116 papers.
Sort:
By area:
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2023A Survey of Large Language Modelscited by 2,537
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2024DeepSeek-V3 Technical Reportcited by 724
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 711
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
- 2025Kimi K2: Open Agentic Intelligencecited by 523
- 2022DeepSpeed Inference: Enabling Efficient Inference of Transformer Models at Unprecedented Scalecited by 403
- 20218-bit Optimizers via Block-wise Quantizationcited by 395
- 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 339
- 2021ZeRO-Infinity: Breaking the GPU Memory Wall for Extreme Scale Deep Learningcited by 253
38 more
- 2023A Survey of Large Language Models for Healthcare: from Data, Technology, and Applications to Accountability and Ethicscited by 162
- 2022A Review of Sparse Expert Models in Deep Learningcited by 147
- 2021Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Modelscited by 78
- 2021Scalable and Efficient MoE Training for Multitask Multilingual Modelscited by 76
- 2021TeraPipe: Token-Level Pipeline Parallelism for Training Large-Scale Language Modelscited by 74
- 2023Advancing Transformer Architecture in Long-Context Large Language Models: A Comprehensive Surveycited by 71
- 2025Kling-Omni Technical Reportcited by 60
- 2025Vchitect-2.0: Parallel Transformer for Scaling Up Video Diffusion Modelscited by 58
- 2023ZeRO++: Extremely Efficient Collective Communication for Giant Model Trainingcited by 50
- 2023ASTRA-sim2.0: Modeling Hierarchical Networks and Disaggregated Systems for Large-model Training at Scalecited by 47
- 2021ERNIE 3.0 Titan: Exploring Larger-scale Knowledge Enhanced Pre-training for Language Understanding and Generationcited by 45
- 2022PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communicationcited by 45
- 2022BNS-GCN: Efficient Full-Graph Training of Graph Convolutional Networks with Partition-Parallelism and Random Boundary Node Samplingcited by 43
- 2023Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuningcited by 39
- 2023Zero Bubble Pipeline Parallelismcited by 39
- 2024ReaL: Efficient RLHF Training of Large Language Models with Parameter Reallocationcited by 34
- 2024TextHawk2: A Large Vision-Language Model Excels in Bilingual OCR and Grounding with 16x Fewer Tokenscited by 32
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Surveycited by 25
- 2021Maximizing Parallelism in Distributed Training for Huge Neural Networkscited by 23
- 2024A Survey: Collaborative Hardware and Software Design in the Era of Large Language Modelscited by 21
- 2024QSpec: Speculative Decoding with Complementary Quantization Schemescited by 18
- 2021Persia: An Open, Hybrid System Scaling Deep Learning-based Recommenders up to 100 Trillion Parameterscited by 17
- 2025Taming the Titans: A Survey of Efficient LLM Inference Servingcited by 16
- 2021Themis: A Network Bandwidth-Aware Collective Scheduling Policy for Distributed Training of DL Modelscited by 15
- 2021Citadel: Protecting Data Privacy and Model Confidentiality for Collaborative Learning with SGXcited by 14
- 2023Quantized Distributed Training of Large Models with Convergence Guaranteescited by 13
- 2023An Adaptive Placement and Parallelism Framework for Accelerating RLHF Trainingcited by 13
- 2024Pipeline Parallelism with Controllable Memorycited by 11
- 2021AdaSplit: Adaptive Trade-offs for Resource-constrained Distributed Deep Learningcited by 8
- 2022HammingMesh: A Network Topology for Large-Scale Deep Learningcited by 8
- 2024Domino: Eliminating Communication in LLM Training via Generic Tensor Slicing and Overlappingcited by 8
- 2023Accelerating Distributed ML Training via Selective Synchronizationcited by 7
- 2024Model Parallelism on Distributed Infrastructure: A Literature Review from Theory to LLM Case-Studiescited by 7
- 2024Demystifying the Communication Characteristics for Distributed Transformer Modelscited by 7
- 2024Hardware Scaling Trends and Diminishing Returns in Large-Scale Distributed Trainingcited by 7
- 2021Automatic Graph Partitioning for Very Large-scale Deep Learningcited by 6
- 2021Making EfficientNet More Efficient: Exploring Batch-Independent Normalization, Group Convolutions and Reduced Resolution Trainingcited by 6
- 2024Activations and Gradients Compression for Model-Parallel Trainingcited by 6
50 most cited of 116 All 116 (JSON)
- 2026The Life of a Token: from Words to Bits on the Wire17 Sep 2026
- 20264D Parallelism Unlocks Exascale Bayesian Neural Networks for High-Fidelity Atmospheric Modeling11 Sep 2026
- 2026KV-Pipe: On the Relation Between KV Sharing and Pipeline Parallel Efficiency in LLMs16 Aug 2026
- 2026RAG-Stack: Co-Optimizing RAG Serving Performance and Quality4 Aug 2026
- 2026Auto-Scaling Heterogeneous Neural Processing Units for Energy and Cost-Efficient LLM Serving17 Jul 2026
- 2026Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency5 Jun 2026
- 2026Demystifying Pipeline Parallelism: First Theory for PipeDream2 Jun 2026
- 2026Characterization of Multi-Model Agentic AI Systems on General Tasks via Trace-Driven Simulation1 Jun 2026
- 2026Unextractable Protocol Models: Collaborative Training and Inference without Weight Materialization22 May 2026
- 2026boldsymbolf-OPD: Stabilizing Long-Horizon On-Policy Distillation with Freshness-Aware Control18 May 2026
- 2026Efficient Training on Multiple Consumer GPUs with RoundPipe29 Apr 2026
- 2026Parallelization Strategies for Dense LLM Deployment: Navigating Through Application-Specific Tradeoffs and Bottlenecks5 Mar 2026
38 more
- 2025Mesh-Attention: A New Communication-Efficient Distributed Attention with Improved Data Locality24 Dec 2025
- 2025iOS as Acceleration19 Dec 2025
- 2025Kling-Omni Technical Report18 Dec 2025
- 2025Scalable Synthesis of distributed LLM workloads through Symbolic Tensor Graphs13 Nov 2025
- 2025CLM: Removing the GPU Memory Barrier for 3D Gaussian Splatting7 Nov 2025
- 2025AsyncHZP: Hierarchical ZeRO Parallelism with Asynchronous Scheduling for Scalable LLM Training23 Oct 2025
- 2025First Attentions Last: Better Exploiting First Attentions for Efficient Transformer Training16 Oct 2025
- 2025OptPipe: Memory- and Scheduling-Optimized Pipeline Parallelism for LLM Training6 Oct 2025
- 2025SlimPack: Fine-Grained Asymmetric Packing for Balanced and Efficient Variable-Length LLM Training30 Sep 2025
- 2025Go With The Flow: Churn-Tolerant Decentralized Training of Large Language Models25 Sep 2025
- 2025Characterizing the Efficiency of Distributed Training: A Power, Performance, and Thermal Perspective12 Sep 2025
- 2025SpikingBrain: Spiking Brain-inspired Large Models5 Sep 2025
- 2025Tri-Accel: Curvature-Aware Precision-Adaptive and Memory-Elastic Optimization for Efficient GPU Usage23 Aug 2025
- 2025WeChat-YATT: A Scalable, Simple, Efficient, and Production Ready Training Library11 Aug 2025
- 2025G-Core: A Simple, Scalable and Balanced RLHF Trainer30 Jul 2025
- 2025Kimi K2: Open Agentic Intelligence28 Jul 2025
- 2025Communication-Efficient Distributed Training for Collaborative Flat Optima Recovery in Deep Learning27 Jul 2025
- 2025MPipeMoE: Memory Efficient MoE for Pre-trained Models with Adaptive Pipeline Parallelism27 Jun 2025
- 2025BestServe: Serving Strategies with Optimal Goodput in Collocation and Disaggregation Architectures6 Jun 2025
- 2025DOPPLER: Dual-Policy Learning for Device Assignment in Asynchronous Dataflow Graphs29 May 2025
- 2025Balanced and Elastic End-to-end Training of Dynamic LLMs20 May 2025
- 2025Taming the Titans: A Survey of Efficient LLM Inference Serving28 Apr 2025
- 2025gLLM: Global Balanced Pipeline Parallelism System for Distributed LLM Serving with Token Throttling21 Apr 2025
- 2025Nonuniform-Tensor-Parallelism: Mitigating GPU failure impact for Scaled-up LLM Training8 Apr 2025
- 2025PipeOffload: Improving Scalability of Pipeline Parallelism with Memory Optimization3 Mar 2025
- 2025SkipPipe: Partial and Reordered Pipelining Framework for Training LLMs in Heterogeneous Networks27 Feb 2025
- 2025Vchitect-2.0: Parallel Transformer for Scaling Up Video Diffusion Models14 Jan 2025
- 2025MixGCN: Scalable GCN Training by Mixture of Parallelism and Mixture of Accelerators3 Jan 2025
- 2024DeepSeek-V3 Technical Report27 Dec 2024
- 2024Priority-Aware Model-Distributed Inference at Edge Networks16 Dec 2024
- 2024Hydraulis: Balancing Large Transformer Model Training via Co-designing Parallel Strategies and Data Assignment10 Dec 2024
- 2024Hardware Scaling Trends and Diminishing Returns in Large-Scale Distributed Training20 Nov 2024
- 2024QSpec: Speculative Decoding with Complementary Quantization Schemes15 Oct 2024
- 2024A Survey: Collaborative Hardware and Software Design in the Era of Large Language Models8 Oct 2024
- 2024TextHawk2: A Large Vision-Language Model Excels in Bilingual OCR and Grounding with 16x Fewer Tokens7 Oct 2024
- 2024Domino: Eliminating Communication in LLM Training via Generic Tensor Slicing and Overlapping23 Sep 2024
- 2024Demystifying the Communication Characteristics for Distributed Transformer Models19 Aug 2024
- 2024LoCo: Low-Bit Communication Adaptor for Large-scale Model Training5 Jul 2024
50 newest of 116 All 116 (JSON)
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2022A Review of Sparse Expert Models in Deep Learningcited by 147
- 2021Scalable and Efficient MoE Training for Multitask Multilingual Modelscited by 76
- 2023Pipeline MoE: A Flexible MoE Implementation with Pipeline Parallelismcited by 4
- 2025MPipeMoE: Memory Efficient MoE for Pre-trained Models with Adaptive Pipeline Parallelismcited by 1
- 2025AsyncHZP: Hierarchical ZeRO Parallelism with Asynchronous Scheduling for Scalable LLM Trainingcited by 0
- 2025AsyncHZP: Hierarchical ZeRO Parallelism with Asynchronous Scheduling for Scalable LLM Training23 Oct 2025
- 2025MPipeMoE: Memory Efficient MoE for Pre-trained Models with Adaptive Pipeline Parallelism27 Jun 2025
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Models11 Jan 2024
- 2023Pipeline MoE: A Flexible MoE Implementation with Pipeline Parallelism22 Apr 2023
- 2022A Review of Sparse Expert Models in Deep Learning4 Sep 2022
- 2021Scalable and Efficient MoE Training for Multitask Multilingual Models22 Sep 2021
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan 2021
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding30 Jun 2020
- 2024QSpec: Speculative Decoding with Complementary Quantization Schemescited by 18
- 2025Taming the Titans: A Survey of Efficient LLM Inference Servingcited by 16
- 2025gLLM: Global Balanced Pipeline Parallelism System for Distributed LLM Serving with Token Throttlingcited by 4
- 2025Balanced and Elastic End-to-end Training of Dynamic LLMscited by 0
- 2026Auto-Scaling Heterogeneous Neural Processing Units for Energy and Cost-Efficient LLM Servingcited by 0
- 2026KV-Pipe: On the Relation Between KV Sharing and Pipeline Parallel Efficiency in LLMscited by 0
- 2026KV-Pipe: On the Relation Between KV Sharing and Pipeline Parallel Efficiency in LLMs16 Aug 2026
- 2026Auto-Scaling Heterogeneous Neural Processing Units for Energy and Cost-Efficient LLM Serving17 Jul 2026
- 2025Balanced and Elastic End-to-end Training of Dynamic LLMs20 May 2025
- 2025Taming the Titans: A Survey of Efficient LLM Inference Serving28 Apr 2025
- 2025gLLM: Global Balanced Pipeline Parallelism System for Distributed LLM Serving with Token Throttling21 Apr 2025
- 2024QSpec: Speculative Decoding with Complementary Quantization Schemes15 Oct 2024
- 2024DeepSeek-V3 Technical Reportcited by 724
- 2024DeepSeek-V3 Technical Report27 Dec 2024
- 2023A Survey of Large Language Modelscited by 2,537
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 711
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
- 2022DeepSpeed Inference: Enabling Efficient Inference of Transformer Models at Unprecedented Scalecited by 403
- 20218-bit Optimizers via Block-wise Quantizationcited by 395
- 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 339
- 2021ZeRO-Infinity: Breaking the GPU Memory Wall for Extreme Scale Deep Learningcited by 253
- 2023A Survey of Large Language Models for Healthcare: from Data, Technology, and Applications to Accountability and Ethicscited by 162
- 2021Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Modelscited by 78
- 2021TeraPipe: Token-Level Pipeline Parallelism for Training Large-Scale Language Modelscited by 74
- 2023Advancing Transformer Architecture in Long-Context Large Language Models: A Comprehensive Surveycited by 71
- 2025Kling-Omni Technical Reportcited by 60
12 most cited of 91 All 91 (JSON)
- 2026The Life of a Token: from Words to Bits on the Wire17 Sep 2026
- 20264D Parallelism Unlocks Exascale Bayesian Neural Networks for High-Fidelity Atmospheric Modeling11 Sep 2026
- 2026Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency5 Jun 2026
- 2026Demystifying Pipeline Parallelism: First Theory for PipeDream2 Jun 2026
- 2026Unextractable Protocol Models: Collaborative Training and Inference without Weight Materialization22 May 2026
- 2026boldsymbolf-OPD: Stabilizing Long-Horizon On-Policy Distillation with Freshness-Aware Control18 May 2026
- 2026Efficient Training on Multiple Consumer GPUs with RoundPipe29 Apr 2026
- 2026Parallelization Strategies for Dense LLM Deployment: Navigating Through Application-Specific Tradeoffs and Bottlenecks5 Mar 2026
- 2025Mesh-Attention: A New Communication-Efficient Distributed Attention with Improved Data Locality24 Dec 2025
- 2025iOS as Acceleration19 Dec 2025
- 2025Kling-Omni Technical Report18 Dec 2025
- 2025Scalable Synthesis of distributed LLM workloads through Symbolic Tensor Graphs13 Nov 2025
12 newest of 91 All 91 (JSON)