InfoQ 中文T2·78 分
Google 与 Meta 推出 TorchTPU 让 PyTorch 原生运行于 TPU
原标题:Google 终于让 PyTorch 原生跑上 TPU,开发者:“早该如此!”
- TorchTPU 由 Google 与 Meta 合作开发,以即时执行为最高优先级,取代需侵入式改代码的 PyTorch/XLA
- 迁移成本大幅降低,仅需替换设备标识与分布式后端,并可用 torch.accelerator 统一识别加速器
- 新增 DeferAndFuse 混合执行模式,Google 称其比 Strict Eager 快 50% 以上