跳到正文
MarkTechPost· Asif Razzaq·· 21 小时前AI 评分73

Google 与 Meta 开源 TorchTPU 原生 PyTorch TPU 后端

Google Open-Sources TorchTPU, a Native PyTorch Backend for Cloud TPUs

AI 导读

Google 与 Meta PyTorch 团队开源 TorchTPU,使 PyTorch 模型可通过 torch.device("tpu") 原生运行在 Google Cloud TPU 上。

来源:MarkTechPost · marktechpost.com