Chengshu@skadai · 2026.10.03
RSS

如何扩展你的模型(How To Scale Your Model)

Google DeepMind 的 LLM 扩展书全书中英对照:左栏原文,右栏译文

以 Google DeepMind 开源书《How To Scale Your Model: A Systems View of LLMs on TPUs》(jax-ml/scaling-book)为底稿的中英对照全文翻译。从 roofline 分析与 TPU 硬件讲起,覆盖分片矩阵乘法、Transformer 数学、FSDP/Megatron/流水线并行、训练与推理的成本估算、Profiling、JAX 编程,以及附赠的 GPU 章节。原文 95 张配图全部保留;roofline、strong scaling、ICI、MXU、KV cache、FSDP 等不好直译的术语保留英文写法。每篇都是左右对照:左栏英文原文,右栏对应中文译文。

目录

已更新 4 篇
  1. 第 0 篇如何扩展你的模型(0):引言
  2. 第 1 篇如何扩展你的模型(1):Roofline 分析入门
  3. 第 2 篇如何扩展你的模型(2):认识 TPU
  4. 第 3 篇如何扩展你的模型(3):分片矩阵与分片矩阵乘法
  5. 第 4 篇如何扩展你的模型(4):Transformer 数学全知道待更新
  6. 第 5 篇如何扩展你的模型(5):如何并行训练一个 Transformer待更新
  7. 第 6 篇如何扩展你的模型(6):在 TPU 上训练 LLaMA 3待更新
  8. 第 7 篇如何扩展你的模型(7):Transformer 推理全知道待更新
  9. 第 8 篇如何扩展你的模型(8):在 TPU 上部署 LLaMA 3待更新
  10. 第 9 篇如何扩展你的模型(9):如何给 TPU 代码做性能剖析待更新
  11. 第 10 篇如何扩展你的模型(10):用 JAX 给 TPU 编程待更新
  12. 第 11 篇如何扩展你的模型(11):结论与延伸阅读待更新
  13. 第 12 篇如何扩展你的模型(12):认识 GPU待更新