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/
