PyTorch/XLA

Google 与 Meta 联合开源 TorchTPU，为 TPU 提供原生 PyTorch 后端，取代此前侵入式修改的 PyTorch/XLA 方案。开发者只需将 cuda 设备替换为 tpu、nccl 后端替换为 tpu_dist，即可让现有工作负载在 TPU 上运行，并支持即时执行、编译执行与分布式计算。项目仍处早期，路线图涵盖动态形状、算子覆盖、强化学习与量化等方向。