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

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