TorchTPU开源原生支持PyTorch

Google 发布开源 PyTorch 专用版本 TorchTPU，帮助开发者将现有模型和应用迁移到 TPU，而无需重写底层代码。Google 还推出 TPU agent suite 自动处理迁移错误，并更新 LiteRT 以兼容基于 PyTorch 的应用；其相关功能与代码库已适配 Meta、Hugging Face、vLLM 和 SGLang 等开发者工具。

Google 开源了 PyTorch 的 TPU 适配项目 TorchTPU，开发者只需修改配置，无需重写应用，即可将现有 AI 应用迁移到 TPU 和 Google Cloud。Google 还推出 TPU agent suite 协助迁移和排错，并更新 LiteRT 以提升其对 PyTorch 应用的兼容性。

Google 开源 TorchTPU，为 PyTorch 模型和操作提供原生 TPU 后端。当前 PyPI 公共版本为 0.2.0，支持 Python 3.11+，并包含 ATen 内核、XLA 编译缓存、torch.compile() 图模式及分布式训练示例。项目采用 Apache 2.0 许可证，模型示例涵盖 Llama、Qwen、ResNet、DLRM、minGPT 和 UNet。

Google 与 Meta PyTorch 团队开源 TorchTPU，使 PyTorch 模型可通过 torch.device("tpu") 原生运行在 Google Cloud TPU 上。

Hugging Face 发布 TorchTPU，将 TPU 集成到 PyTorch，并让 Transformers、Diffusers 和 Kernels 可原生运行在 TPU 上。