谷歌联手 Meta 开源 TorchTPU,改一行代码即可在 TPU 上跑 PyTorch
lmoroney · x · 2026-10-21
Google 与 Meta 在 PyTorch Conference 上宣布开源 TorchTPU——一个原生 PyTorch TPU 后端，项目于 2025 年中在 Google 启动。目标是将迁移成本降到把 .to("cuda") 改成 .to("tpu") 的程度。
要点：
- 与 2020 年基于 Lazy Tensor 的 PyTorch/XLA 不同，旧方案需要侵入式修改源码、编译延迟高、不支持 eager 模式；TorchAX 等方案则是把 PyTorch 塞进 JAX 的模式。
- TorchTPU 以 eager 执行为最高优先级，同时支持 torch.compile，覆盖约 450 个 ATen 算子，现有 PyTorch 工作负载可开箱即用地迁移到 TPU。
- 团队称推理与训练性能达到业界竞争力水平，底层仍通过 XLA 编译栈驱动 TPU。
所属事件:谷歌联手 Meta 开源 TorchTPU，PyTorch 原生跑上 TPU(15 条相关)→
「Infra」频道最新
- pi 模型推理引擎 rec 两行代码即可在 M5 Max 本地运行 — _rockt · 2026-10-22
- 安全研究员警告:GitLab 漏洞严重,呼吁用户尽快打补丁 — basedjensen · 2026-10-22
- Meta 财报：营收 646.1 亿美元增 26%，2026 资本开支上调至 1350-1450 亿美元 — RihardJarc · 2026-10-22
- PyTorch 大会 Demo Theater 13 场演示:Wan 2.2 视频生成提速 53.6 倍 — PyTorch · 2026-10-22
- TorchTitan 大更新:支持 DeepSeek-V4 等 5D 并行,Muon 优化提速 79% — PyTorch · 2026-10-22
- JNTC 将 TGV 玻璃基板铜填充时间从一天缩至分钟,明年量产玻璃芯 — toptickcrypto · 2026-10-22