Google 与 Meta 联合推出 TorchTPU：PyTorch 原生 TPU 后端

在 PyTorchCon 2026 上，Google Cloud 的 Bill Jia 和 Meta 的 Zongwei Zhou 发布了 TorchTPU，一个面向 Google TPU 的 PyTorch 原生后端。TorchTPU 由 Google 和 Meta 开源合作开发，与 PyTorch 核心维护者密切协作。它已集成 Hugging Face Transformers 用于建模、TorchTitan 用于训练、vLLM 和 SGLang 用于推理服务。演示中还展示了如何用智能体将复杂模型工作负载从 GPU 迁移到 TPU，并利用智能体做量化、自定义内核生成和分片策略优化。