跳到正文
MarkTechPost· Asif Razzaq·· 26 天前AI 评分34

Google 开源 TorchTPU:面向 Cloud TPU 的原生 PyTorch 后端

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

AI 导读

Google 开源了与 Meta PyTorch 团队联合开发的 TorchTPU,一个让 PyTorch 模型原生运行在 Cloud TPU 上的后端,取代 PyTorch/XLA,PyPI 版本 v0.2.0、Apache 2.0 许可。

来源:MarkTechPost · marktechpost.com