Google open-sources TorchTPU, a native PyTorch back end for TPUs

Google 开源 TorchTPU，为 TPU 提供原生 PyTorch 后端，让 PyTorch 模型和算子可直接在 TPU 设备上运行。项目包含自定义 ATen kernel、XLA 计算编译缓存，支持数据并行与张量并行分布式训练，并集成 torch.compile() 图模式执行。