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