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

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

来源与原文

AIHOT · 精选

作者:noreply@aihot.news (PyTorch:Blog)

发布时间:(北京时间)

阅读原始来源 · aihot.news ↗
完整链接与内容说明

https://aihot.news/items/lh8vm7yp332xoffhqoscusss4

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

Google 与 Meta 联合开源 TorchTPU,为 Google TPU 提供原生 PyTorch 后端。模型只需把 `.to("cuda")` 改成 `.to("tpu")` 即可开箱运行,无需 JAX 依赖。

来源:AIHOT · 精选 · aihot.news