Google 开源 TorchTPU，让 PyTorch 原生跑在 TPU 芯片上

Google 发布 TorchTPU，这是 PyTorch 的原生后端，可让 PyTorch 代码直接在 TPU 芯片上以完整性能运行，思路类似 Amazon 让 PyTorch 编译到 Trainium 芯片的 TorchNeuron。该项目由 Google 与 Meta 的 PyTorch 团队共同开发，2025 年启动，并在 PyTorch Conference 2026 上以全开源形式上架 GitHub。Google 还与 Hugging Face 合作测试多个开源模型在 TorchTPU 上的表现，并联合 vLLM、SGLang、Ray、TorchTitan 等库验证 TPU 上的实际使用。