跳到正文
PyTorch:Blog·· 13 天前AI 评分74

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

Introducing TorchTPU

AI 导读

Google 与 Meta 联合开源 TorchTPU,为 Google TPU 提供原生 PyTorch 后端,模型只需把 .to("cuda") 改成 .to("tpu") 即可开箱运行,无需 JAX 依赖。

来源:PyTorch:Blog · pytorch.org