跳到正文
原文
AIHOT · AI 精选· noreply@aihot.news (PyTorch:Blog)·· 2 小时前精选AI 评分87

Google 与 Meta 联合发布 TorchTPU:原生 PyTorch TPU 后端

Google 与 Meta 发布 TorchTPU:PyTorch 原生 TPU 后端

AI 导读

Google 与 Meta 联合发布开源项目 TorchTPU,为 Google TPU 提供原生 PyTorch 后端。该工具支持通过修改设备参数(如将 .to("cuda") 改为 .to("tpu"))实现模型无缝迁移,并兼容 eager 执行模式与 torch.compile。

推荐理由

原文给出了从GPU迁移到TPU的具体代码改动、精度策略和分布式支持细节,品牌可据此评估现有PyTorch工作流迁移至TPU的改动成本与性能收益。

来源:AIHOT · AI 精选 · aihot.news