系统讲解大模型训练加速的核心算法与技术,涵盖计算优化(Flash Attention 的演进、DS GEMM 稀疏计算与 MTP 并行)、序列优化(Ulysses 多流调度与 Ring Attention 长序列处理)、内存优化(MLA 注意力机制与梯度检查点)、混合精度训练(FP8 表示与稳定性)以及通算融合(DeepSpeed-Domini 与 FLUX 的通信计算重叠)等关键方向,旨在通过原理剖析与实现细节全面了解如何提升大模型训练效率。
建议优先下载 PDF 版本,PPT 版本会因为字体缺失等原因导致版本很丑哦~
| 大纲 | 小节 | 链接 | 状态 |
|---|---|---|---|
| 大模型训练加速 | PPT, 文章, 视频 | ||
| 💖 | 🌟 | 💖 | |
| 并行 实践 💻 | CODE 01: Flash Attention 实现 | Markdown, Jupyter, 文章 | ✅ |
| 并行 实践 💻 | CODE 02: 梯度检查点内存优化 | Markdown, Jupyter, 文章 | ✅ |
| 并行 实践 💻 | CODE 03: FP8 混合精度训练 | Markdown, Jupyter, 文章 | ✅ |
| 并行 实践 💻 | CODE 04: Ring Attention 实践 | Markdown, Jupyter, 文章 | ✅ |
文字课程内容正在一节节补充更新,每晚会抽空继续更新正在 AI Infra ,希望您多多鼓励和参与进来!!!