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