跳到正文
TechCrunch · AI· Julie Bort·· 13 小时前精选AI 评分78

Google 开源 TorchTPU,让 PyTorch 应用无需重写即可跑在 TPU 上

Google has joined Amazon in attacking Nvidia’s stronghold with AI app developers

AI 导读

Google 周二宣布开源 TorchTPU,这是 PyTorch 的 TPU 版本,开发者只需改配置、无需重写应用,就能把现有 PyTorch 应用迁移到 Google 的 TPU 和 Google Cloud。

推荐理由

Google 与 Amazon 先后推出 PyTorch 的 TPU 与 Trainium 适配版本,读者可据此观察非 Nvidia 硬件生态的推进方式。

来源:TechCrunch · AI · techcrunch.com