AI圈报
产品发布 / 更新普通

Google 开源 TorchTPU,为 TPU 提供原生 PyTorch 后端

信息来源:X:Rohan Paul (@rohanpaul_ai)·
原始标题:Google and Amazon now both offer native PyTorch backends that let code leave Nvidia GPUs.

内容摘要

Google 开源 TorchTPU,为 TPU 提供原生 PyTorch 后端,替代此前的 PyTorch/XLA 层。它按写入顺序逐个运行算子以便像 GPU 一样逐步调试,再切换 torch.compile 获得生产速度;训练 Qwen 3 8B 时最高用到 52% 的峰值 TPU 算力。原文还提到 Amazon 也提供原生 PyTorch 后端,让代码脱离 Nvidia GPU 运行。
内容层级普通情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源X:Rohan Paul (@rohanpaul_ai)
站内情报编号intel-02ee18de242eec692a68e230