2026 年 10 月 20 日,Google 将 TorchTPU 在 GitHub 开源。它为 Google TPU 提供原生 PyTorch 后端,开发者可使用熟悉的 PyTorch API 训练和部署模型,并通过 eager mode 调试、torch.compile 优化执行。

Google 表示,TorchTPU 已与 vLLM、SGLang、TorchTitan、Ray 和 Hugging Face 等工具集成,使现有模型工作流可以接入 TPU。代码仓库已公开,并提供安装及构建说明;具体性能仍取决于模型和部署配置。

来源:Google Cloud 发布说明、TorchTPU 代码仓库。