机器之心· 关注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