谷歌已在 GitHub 开源 TorchTPU,仓库地址为 github.com/google-pytorch/torch_tpu。从项目命名与官方给出的一句话描述来看,其定位是“TPU 的原生 PyTorch 后端”,也就是让 PyTorch 程序能够直接以 TPU 作为执行目标。
对 AI 基础设施从业者来说,这个方向值得关注,原因在于框架与硬件之间的绑定关系。PyTorch 是当前训练与推理的主力框架之一,而 TPU 是谷歌自研的加速器;过去在 TPU 上运行 PyTorch,通常要经过额外的中间层或代码调整,前端体验与 GPU 上的原生路径存在差距。如果“原生后端”按字面落地,就意味着 PyTorch 侧的算子、编译与运行时可以直接对接 TPU,从而减少框架切换与代码迁移成本,也扩大 TPU 对既有 PyTorch 代码库和团队技能栈的可达面。
选择开源形式本身也有信号意义:实现细节、支持范围与限制可被外部开发者直接检视,适配需求能够以 issue、PR 的形式进入上游,而不是停留在封闭的内部工具链中。
需要说明的是,目前可确认的信息仅限于仓库的存在与项目自述,尚无版本号、支持算子范围、性能数据、成熟度或路线图等细节。其实际可用性与生产就绪程度,仍有待后续文档与代码披露。
