跳到正文
掘金· lucas_AI·· 24 天前AI 评分50

Google 与 Meta 开源 TorchTPU,PyTorch 原生 TPU 后端改 device 字符串即可运行

Google 和 Meta 把 TPU 变成 PyTorch「原生设备」:TorchTPU 开源,device="cuda" 改一个词就能上 TPU

AI 导读

Google 与 Meta 在 2026-10-20 的 PyTorch Conference 上联合开源 TorchTPU,一个运行时零 JAX 依赖、eager 优先的 PyTorch 原生 TPU 后端,把 device 从 cuda 改成 tpu、backend 从 nccl 改成 tpu_dist 即可运行。

来源:掘金 · juejin.cn