跳到正文
AIHOT 精选· noreply@aihot.news (PyTorch:Blog)·· 7 天前AI 评分66

Google 与 Meta 联合开源 TorchTPU:PyTorch 原生 TPU 后端

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

AI 导读

Google 与 Meta 合作开源 TorchTPU,为 Google TPU 提供符合 PyTorch 设计原则的原生后端,运行时不依赖 JAX,模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行。

这篇内容暂未取得可阅读的正文,目前提供中文导读和原文入口。节目介绍或原站公开片段不代表完整文章。

来源:AIHOT 精选 · pytorch.org

原文链接:https://pytorch.org/blog/introducing-torchtpu/