InfoQ 中文 · AI· 作者：Andrew Hoblitzell·· 5 天前精选AI 评分78
谷歌和 Meta 开源 TorchTPU，为 TPU 提供原生 PyTorch 后端
2026 年 PyTorch 大会：谷歌和 Meta 开源 TorchTPU，在 TPU 上原生支持 PyTorch
AI 导读
在圣何塞举行的 2026 年 PyTorch 北美大会上，谷歌和 Meta 开源了 TorchTPU，这是 TPU 的原生 PyTorch 后端，取代需要侵入式改代码的 PyTorch/XLA，目标是让模型迁移只需把 model.to("cuda") 改成 model.to("tpu")。
推荐理由
TorchTPU 把 TPU 接入 PyTorch 原生后端，文中给出迁移改动量、扩展数据与调优缺口，便于评估换芯成本。
来源：InfoQ 中文 · AI · infoq.cn