Meta 工程师证实已用 Google TPU 在生产环境跑 AI 推理

dotey · x · 2026-10-21

Google 在开发者博客发布 TPU 版 vLLM 与 SGLang 原生推理引擎,并引用 Meta PyTorch-TPU 负责人 Zongwei Zhou 的原话:“SGLang-TorchTPU 今天已经在 Meta 的生产环境中运行。”这是 Meta 使用 Google TPU 跑生产推理的首次一手确认。SemiAnalysis 早在 2025 年 11 月就报道称 Meta 是 TPU 客户,The Information 今年 2 月报道两家签了多年、数十亿美元的 TPU 租用合同,但 Meta 未公开规模、涉及产品及租买方式。

技术层面:Google 推出 TorchTPU,让 TPU 成为原生 PyTorch 设备,张量以标准 PyTorch 张量管理,支持 eager 执行与 torch.compile(经 XLA 编译),vLLM 与 SGLang 在其上保留各自的 worker 模型、调度器、KV cache 管理与 API server,底层替换为 TPU 实现,并附带优化的 Pallas 内核。此举被视为对英伟达 CUDA 生态壁垒的直接挑战。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →