Google 开源 TorchTPU，为 TPU 提供原生 PyTorch 后端

导读

Google 开源 TorchTPU，为 TPU 提供原生 PyTorch 后端，替代此前的 PyTorch/XLA 层。它按写入顺序逐个运行算子以便像 GPU 一样逐步调试，再切换 torch.compile 获得生产速度；训练 Qwen 3 8B 时最高用到 52% 的峰值 TPU 算力。原文还提到 Amazon 也提供原生 PyTorch 后端，让代码脱离 Nvidia GPU 运行。