Google 开源 TorchTPU：面向 TPU 的原生 PyTorch 后端

Google 开源 TorchTPU，为 TPU 提供原生 PyTorch 后端，让 PyTorch 模型和算子可直接在 TPU 设备上运行。项目包含自定义 ATen kernel、XLA 计算编译缓存，并支持数据并行与张量并行分布式训练，附带 Llama、Qwen、ResNet、DLRM、minGPT、UNet 等示例。