deepseek-ai

DualPipe

deepseek-ai

A bidirectional pipeline parallelism algorithm for computation-communication overlap in DeepSeek V3/R1 training.

AI 简介

DualPipe 是一种面向大模型训练的双向流水线并行算法,旨在实现前向与反向计算和通信的完全重叠,显著减少流水线气泡并提升硬件利用率。其核心特点是通过双向微批次调度(正向与反向交替执行)和计算-通信重叠机制,在保持模型并行扩展性的同时降低通信等待开销;支持 DualPipeV 等衍生精简调度变体。适用于千亿级参数模型在多GPU集群上的高效分布式训练,尤其适合计算与通信带宽比受限的大规模语言模型训练场景。

Python
MIT License
3k
Stars
329
Forks
28
Watchers
3
Issues

Star 增长

今日0
近 7 天0
近 30 天+3
综合评分59.86
默认分支main