Google 开源 TorchTPU：为 TPU 提供 PyTorch 后端

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