TorchTPU

Google 开源项目，让 PyTorch 工作负载原生运行在 TPU 上。

核心特性（2026-04-24 发布）

• 原生 PyTorch 支持 — 无需 XLA 中间层
• 支持 eager execution — 保持 PyTorch 的即时执行模式
• 支持 torch.compile — 兼容 PyTorch 2.x 编译优化
• Google 规模验证 — 在 Google 内部大规模工作负载上测试

意义

降低了从 GPU 迁移到 TPU 的门槛。此前 PyTorch 在 TPU 上需要通过 XLA（Accelerated Linear Algebra）作为中间层，增加了复杂性和调试难度。TorchTPU 消除了这一层，使 TPU 成为 PyTorch 的一等公民。

参见

• Cloud — 云计算与 AI 基础设施