跳到正文
TechCrunch 硬件· Julie Bort·· 3 天前精选AI 评分77

Google 开源 TorchTPU,支持 PyTorch 应用迁移至 TPU

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

AI 导读

Google 将 TorchTPU 作为开源项目发布,使开发者可通过简单配置变更让现有 PyTorch 应用运行在 TPU 和 Google Cloud 上,无需重写代码。Google 还推出 TPU agent suite 协助迁移和排错,并提升 LiteRT 对 PyTorch 应用的兼容性。Amazon 同日发布了面向 Trainium 和 AWS 的类似方案 TorchNeuron。

推荐理由

Google给出了无需重写代码即可将PyTorch应用迁移到TPU和Google Cloud的路径,并补充迁移排错工具与生态兼容信息。

来源:TechCrunch 硬件 · techcrunch.com