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