跳到正文
PyTorch:Blog· noreply@aihot.news (PyTorch:Blog)·· 9 小时前精选

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

AI 导读

Google 与 Meta 发布开源项目 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型迁移只需把 .to("cuda") 改为 .to("tpu")。

推荐理由

来源于 AI Hot 精选订阅

来源:PyTorch:Blog · pytorch.org