Google 与 Meta 开源 TorchTPU,PyTorch 原生跑 TPU 无需改代码
PyTorch · x · 2026-10-23
PyTorchConf 2026 上，Google 与 Meta 官方开源 TorchTPU——PyTorch 在 Google TPU 上的原生后端，由 Google 2025 年中启动、与 Meta PyTorch 团队合作开发。与 2020 年基于 Lazy Tensors 的 PyTorch/XLA 相比，旧方案需要侵入式修改源码、编译延迟高、不支持 eager 模式且分布式原语支持滞后；TorchTPU 则将 eager 执行作为首要目标，集成 torch.compile 并提供原生硬件加速，实现开箱即用的模型执行与 TPU 迁移，训练与推理性能具业界竞争力。
所属事件:Google 联手 Meta 开源 TorchTPU，PyTorch 原生支持 TPU(2 条相关)→
「Infra」频道最新
- 重建 AI Agent 算力栈的三条心得：安全沙箱、近零闲置成本与持久记忆 — Saboo_Shubham_ · 2026-10-23
- Intel CRO 证实与 Musk 合作建 Terafab：芯片需求大到自建晶圆厂 — elonmusk · 2026-10-23
- 研究员吐槽:20万美元GPU服务器的采购单要用信封手递 — HildeKuehne · 2026-10-23
- Intel 与 AMD 合作：oneDNN 成为 PyTorch x86 CPU 加速统一路径 — PyTorch · 2026-10-23
- 同一模型同卡可先卡内存后卡算力：GPU 执行原理全图解 — techNmak · 2026-10-23
- Intel 推 AI Playground Home Agent:手机经 Slack 远程调用本地 AI PC 算力 — PyTorch · 2026-10-23