Google 开源原生 PyTorch TPU 后端 TorchTPU

Google 开源 TorchTPU，为 PyTorch 模型和操作提供原生 TPU 后端。项目包含自定义 ATen 算子、XLA 编译缓存、分布式训练与 torch.compile() 集成，并提供 Llama、Qwen、ResNet 等示例；当前 PyPI 公共版本为 0.2.0，支持 Python 3.11+，采用 Apache 2.0 License。