Google 与 Meta 联合开源 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行,无需 JAX 依赖。 🔗 阅读原文 via AIHOT · https://aihot.news/items/lh8vm7yp332xoffhqoscusss4
信息源:AI HOT · 精选
- 原文链接:Google 与 Meta 发布 TorchTPU:PyTorch 原生 TPU 后端
- 分类:AI 产品
- 作者:PyTorch:Blog
- 发布时间:2026/10/21 00:00

评论0