跳到正文
机器之心· 关注TPU的·· 4 小时前精选AI 评分94

谷歌与 Meta 联合开源 TorchTPU,实现 PyTorch 原生 TPU 支持

谷歌开源TorchTPU!TPU成PyTorch原生设备,CUDA最深的护城河开始松动

AI 导读

谷歌在 PyTorch Conference 2026 宣布开源 TorchTPU,使 TPU 成为 PyTorch 的原生设备,部署只需指定 device_map="tpu"。vLLM、SGLang 等主流框架同步完成接入,Qwen3 235B 模型在 8192 颗 TPU 上扩展效率保持较高。

推荐理由

文章梳理了 TorchTPU 实现 PyTorch 原生 TPU 运行所需的关键架构改动、性能基准和生态接入进展。

来源:机器之心 · mp.weixin.qq.com