在 PyTorch Conference 2026 上,Google 宣布将 TorchTPU 开源。TorchTPU 是 Google 与 Meta PyTorch 团队联合开发的原生 PyTorch 后端,使 TPU 与 GPU、CPU 一样成为 PyTorch 的一等公民设备。开发者只需对现有 PyTorch 代码做最小改动,即可将工作负载迁移到 Google TPU 上运行,继续使用熟悉的 PyTorch API。官方表示,TorchTPU 已在 GitHub 上开放源代码,PyTorch 官方博客也同步发布了介绍。这一举动被认为将降低 PyTorch 生态使用 TPU 的门槛,对 NVIDIA CUDA 生态的主导地位构成挑战。