从零推导构建 PyTorch 分布式训练框架

hkproj · x · 2026-08-18

这是一部极长视频教程,从第一性原理出发使用 PyTorch 构建分布式训练框架。内容包括推导数学原理、Multi-head Latent Attention (MLA)、RoPE、YaRN 等组件。详细讲解进程组、Rank、集合通信、分布式 Autograd、设备网格与张量,并实现流水线并行、数据并行、FSDP、HSDP、张量并行和上下文并行等现代 LLM 训练的核心并行技术。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →