Google 开源 TorchTPU：TPU 的 PyTorch 后端

Google 开源了 TorchTPU，为 TPU 提供 PyTorch 后端，让用户能在 TPU 设备上运行 PyTorch 模型和算子。项目包含 TPU 的 ATen kernel 实现、XLA 计算编译缓存、数据并行与张量并行分布式训练支持，并集成 torch.compile() 图模式执行。 使用条件：目前仅支持 Python 3.11+，需在 TPU VM 上运行（v6e GCE 实例最简便），从 PyPI 安装最新公开发行版 0.2.0；项目采用 Apache 2.0 许可证。

AI 根据报道生成 · 1 小时前更新

还没有足够的连续观测数据，暂不绘制趋势。