跳到正文
SiliconANGLE· Mike Wheatley·· 1 天前精选AI 评分78

Google 发布 TorchTPU,帮助 PyTorch 应用运行在 TPU 上

Google unveils TorchTPU in a push to break Nvidia’s stranglehold on AI compute

AI 导读

Google 发布 TorchTPU,这是面向 TPU 的 PyTorch 专用版本,可通过少量配置调整运行现有应用和模型。Google 还推出 TPU agent suite 以辅助迁移并自动修复错误,同时更新 LiteRT 以兼容基于 PyTorch 的应用。

推荐理由

文章将 TorchTPU 与 AWS 的 TorchNeuron 放在同一背景下,具体呈现 PyTorch 兼容层如何降低 AI 应用迁移到不同硬件的门槛。

来源:SiliconANGLE · siliconangle.com