AI风向标

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

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

详细介绍

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

AI HOT 详情:https://aihot.news/items/lh8vm7yp332xoffhqoscusss4

原文链接:https://pytorch.org/blog/introducing-torchtpu/