知乎专栏：机器之心· 机器之心·· 3 小时前精选AI 评分86
Google 开源 TorchTPU，使 TPU 成为 PyTorch 原生设备
谷歌开源TorchTPU！TPU成PyTorch原生设备，CUDA最深的护城河开始松动
AI 导读
Google 在 PyTorch Conference 2026 上宣布开源 TorchTPU，让 PyTorch 原生运行在 TPU 上；在 Hugging Face Transformers v5 中部署模型时，可通过 device_map="tpu" 指定设备。
推荐理由
文章梳理了 TorchTPU 的原生设备接入、即时执行与编译路径，帮助读者理解 PyTorch 模型迁移到 TPU 时的软件栈变化。
来源：知乎专栏：机器之心 · zhuanlan.zhihu.com