Skip to content
简体中文

切换主题

// ▲ 预训练、缩放与分布式训练

张量并行(Megatron)

把一层的矩阵乘法切到多张卡上同时算。

Tensor Parallelism (Megatron-LM) 2019 高级 核心

通俗解释

把注意力头和前馈层的权重按列/行切开,每张卡算一部分,再用少量通信合并结果。适合同一台机器内高速互联的 GPU。

用在哪里:Megatron-LM、NeMo 等

时间轴 · 前置 → 当前 → 后继

2017 Transformer 与…2018 流水线并行2019 张量并行(Megatron)

01原始论文

Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism

Mohammad Shoeybi, Mostofa Patwary, Raul Puri, et al. · 2019

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

左边先学;右边是学完它之后可以学的。

先学

张量并行(Megatron)

学完可以学

相关:ZeRO / FSDP 数据并行

Transformer 与自注…流水线并行ZeRO / FSDP 数据并行 张量并行(Megatron)

左:前置 · 右:后继 · 虚线:相关。看完整关系图 →

03学习资料

官方文档、图解、课程和参考实现,链接都核对过。

04在课程里哪里学

给 AI 读 / 数据说明

数据更新于:AI 算法图谱与课程体系(北京时间)