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

Google 开源了面向 TPU 设备的原生 PyTorch 后端项目 TorchTPU，允许用户直接在 TPU 上运行 PyTorch 模型与算子。该项目提供了自定义 ATen 算子实现、优化 XLA 计算的编译缓存，并集成了 torch.compile() 图模式与分布式训练支持。