智识花园阅读、实践、留下笔记
产品AI 评分 70/100

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

PyTorch:Blog(RSS)查看原始信源 ↗

AI guide

AI 导读

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

数据来源:AIHOT aihot.news ↗ · 内容版权归原作者所有