跳到正文
TechCrunch · AI· Julie Bort·· 5 天前精选AI 评分76

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

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

AI 导读

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

推荐理由

Google 与 Amazon 先后推出 PyTorch 的 TPU、Trainium 适配版本,读者可据此观察 AI 芯片生态的竞争格局。

来源:TechCrunch · AI · techcrunch.com