Skip to content

11. 延伸阅读

本章给出 Google 案例的一手来源(论文、仓库、博客)、按主题分组,以及与手册其它主题的交叉引用和学习路径建议。所有链接于 2026-07-03 核验。


11.1 核心论文(一手)

主题论文 / 来源链接
TPU v4 可靠性NSDI'24《Resiliency at Scale: Managing Google's TPUv4 ML Supercomputer》, Zu et al.usenix.org/system/files/nsdi24-zu.pdf
Borg 集群管理EuroSys'15《Large-scale cluster management at Google with Borg》research.google/pubs/archive/43438.pdf(用 research.google.com
Pathways 运行时MLSys'22《Pathways: Asynchronous Distributed Dataflow for ML》, arXiv:2203.12533arxiv.org/abs/2203.12533
GSPMD 自动分片arXiv:2105.04663《GSPMD: General and Scalable Parallelization for ML Computation Graphs》arxiv.org/abs/2105.04663
TPU v1 架构Jouppi et al.《In-Datacenter Performance Analysis of a Tensor Processing Unit》arxiv.org/abs/1704.04760
TPU v4 架构(含 SparseCore/OCS)ISCA'23《TPU v4: An Optically Reconfigurable 3D Torus ...》arxiv.org/abs/2304.01433
Gemma 3 技术报告arXiv:2503.19786arxiv.org/html/2503.19786v1
Falcon 硬件传输SIGCOMM'25(最佳论文)dl.acm.org/doi/10.1145/3718958.3754353(ACM 付费墙;作者副本 praveenk.io/papers/falcon.pdf
Frontier Safety 评估套件Phuong et al.《Evaluating frontier models for dangerous capabilities》, 2024DeepMind 技术报告

11.2 官方博客与文档

主题来源链接
Multislice 跨 Pod 训练Cloud TPU Multislice 博客cloud.google.com/blog/.../using-cloud-tpu-multislice-...
Falcon 传输Google Systems 博客cloud.google.com/blog/topics/systems/introducing-falcon-...
TPU 架构文档Cloud TPU 性能/架构文档cloud.google.com/tpu/docs
TPU 产品代次(v1–Ironwood)Google Cloud / DeepMind 博客cloud.google.com/blog 搜 "Ironwood" / "Trillium"
投机解码(DFlash,~3×)Google Developers 博客(2026-05)developers.googleblog.com/.../supercharging-llm-inference-...-speculative-decoding
JAX 分布式数组与自动并行JAX 官方文档docs.jax.dev/.../Distributed_arrays_and_automatic_parallelization.html
Gemini API / Vertex AIai.google.dev / cloud.google.com/vertex-ai

11.3 可读源码(开源仓库)

项目仓库用途
MaxTextAI-Hypercomputer/maxtext最易读的 TPU 训练参考实现(JAX);本案例 第 6 章 的主要源码来源
JAXjax-ml/jax函数式可追踪前端;GSPMD 的载体
XLA / OpenXLAopenxla/xla编译器(含 GSPMD 分片 pass)
JetStream / tpu-inferenceAI-Hypercomputer/JetStreamtpu-inferenceTPU 原生推理引擎
MaxDiffusionAI-Hypercomputer/maxdiffusion扩散模型训练参考
GemmaHF google/gemma-3 / ai.google.dev/gemma开放权重模型
Pax / Paxmlgoogle/paxml大规模训练框架(维护/参考角色)
T5Xgoogle-research/t5xT5 风格 encoder-decoder 训练
Orbax / Optax / Flax / Grain / Tunixgoogle/orbaxMaxText 组合使用的依赖库

Pathways 运行时未开源(内部系统,协调底座 PLAQUE 闭源)。


11.4 可持续性、安全与治理

主题来源链接
数据中心可持续(PUE/CFE)Google Data Centersdatacenters.google/operating-sustainably
运营可持续sustainability.googlesustainability.google/operations
AI PrinciplesGoogle AIai.google/principles
Frontier Safety FrameworkDeepMind 博客deepmind.google/blog/introducing-the-frontier-safety-framework

11.5 与手册其它主题的交叉引用

主题关联链接
AI SRENSDI'24 双路径恢复、healthd、preflight、99.98% 可用率是 AI SRE 硬核案例/07-ai-sre/
AI 平台 / 算力底座TPU + GCP 是三大算力底座之一;Multislice 是跨 Pod 弹性算力代表/03-ai-platform/
LLMOps / 推理服务JetStream/tpu-inference 是 TPU 原生推理;投机解码是推理加速代表/04-llmops/
RAG / 向量检索SparseCore 为稀疏嵌入加速,与大规模检索 workload 相关/06-rag/
安全Frontier Safety Framework、AI Principles、PSP 传输加密/08-security/
Meta 案例对照"开放权重 + OCP 硬件 + PyTorch" vs "开放软件栈 + 自研 TPU + JAX"/09-case-study/meta/
OpenAI 案例对照"闭源规模 + 通用 GPU" vs "垂直整合自研硅 + 软件全开"/09-case-study/openai/
Anthropic 案例对照"闭源对齐" vs "开放生态 + 安全框架"/09-case-study/anthropic/

11.6 建议学习路径

  1. 先读 NSDI'24PDF):它是理解 Google 大规模训练可靠性的核心,OCS/3D-torus/双路径恢复都在里有量化数据。
  2. 跑本案例的 Mini Demo:亲手调整 torus 维度、故障率、reroute 步时惩罚,建立对"延迟项"与"双路径权衡"的直觉。
  3. 读 MaxText 源码:从 src/maxtext/utils/sharding.py(分片标注)与 trainers/pre_train/train.py(训练循环)入手,理解"标注即并行"如何落地。
  4. 读 Pathways + GSPMD 论文:理解运行时与编译器的分工(标注 → 分片 → 编排)。
  5. 对照 Meta 案例:把两个案例的"可靠性工程"(NSDI 双路径 vs SDC 治理)放一起,得到完整的"AI 数据中心可靠性"设计图。
  6. 关注可持续与安全治理:PUE 1.09、24/7 CFE、Frontier Safety Framework 是生产落地的硬约束。

11.7 未决/需进一步核实项

为保持严谨,本案例中以下项标注了来源限制(详见各章脚注):

  • TPU 制程节点(28/16/7/4/3 nm):Google 主文档未披露,来自二手来源(Wikipedia/SemiAnalysis)。
  • Ironwood ICI:Google 写 "1.2 TBps bidirectional"(字节、双向),= 9.6 Tbps 双向 / 4.8 Tbps 单向——勿误读为"每链路 1.2 Tbps"。
  • Gemma 3 "N 种语言":技术报告未在正文给出单一数字,仅称"扩大了多语言覆盖"。
  • Frontier Safety Framework v2.0 具体变更:v2.0 PDF 未从博客 HTML 可达。
  • 当前 24/7 CFE 百分比:需查年度环境报告 PDF,页面未直接给数字。
  • Gemini 具体训练/服务 ops 数字:非公开,本案例未引用具体值。
  • NSDI'24 术语:论文无 "cherry-pick"/"performance health"/"functional health";本案例一律用论文真实术语(end-to-end check / intent-driven checker / reconfigure / fault-tolerant ICI routing),详见 第 2 章第 8 章

如发现链接失效或数字有更新,欢迎在仓库提 issue 修正。本案例的开放姿态本身就是"可学习、可纠错"的——正如 Google 开放其软件栈的初衷。

Released under CC-BY-SA-4.0 License.