TorchTPU 是谷歌在 GitHub 开源的 PyTorch TPU 后端项目，目标是让 PyTorch 用户更直接、高效地利用 Google Cloud TPU 算力。该项目通过自定义 ATen 内核适配 TPU 执行特性，引入 XLA 编译缓存以减少重复编译开销，并与 torch.compile 接口集成，使模型在 TPU 上的编译和训练流程更加平滑。仓库还提供分布式训练支持、调试工具，以及 Llama、Qwen 和 ResNet 等主流模型的示例，覆盖大语言模型与视觉模型场景。对关注大规模训练效率、希望降低 TPU 使用门槛的团队而言，这一项目提供了更接近原生 PyTorch 体验的路径，也有助于推动 TPU 生态在开源社区中的扩展。