Google与Meta开源TorchTPU：TPU获原生PyTorch支持
原标题：PyTorch Conference 2026: Google and Meta Open-Source TorchTPU for Native PyTorch on TPUs
AI 摘要
在2026年北美PyTorch大会上，Google与Meta开源了TorchTPU，为Google TPU提供原生PyTorch后端，取代需要修改源码的PyTorch/XLA，使模型迁移简化为将设备切换为TPU。Meta计划借此把更大规模LLM负载迁移到TPU，vLLM、SGLang、TorchTitan等已原生支持，并展示了超8000芯片的近线性扩展。Google同时预告TPU agent套件以自动化调优，AWS也推出Trainium的TorchNeuron后端，但分析指出厂商对优化与供给层的控制意味着锁定并未消失。
正文节选
At PyTorch Conference North America 2026 in San Jose, Google and Meta open-sourced TorchTPU, a native PyTorch backend for Google's Tensor Processing Units (TPUs). Meta, which created PyTorch in 2016, plans to move larger-scale LLM workloads onto TPUs with it, Meta software engineer Zongwei Zhou said in the October 21 keynotes. TorchTPU replaces PyTorch/XLA, a lazy-tensor bridge from 2020 that, according to the announcement, "required intrusive source-code modifications." The new backend aims to