谷歌于当地时间周二宣布,面向谷歌TPU的原生PyTorch框架TorchTPU已在GitHub开源。该框架可让AI应用在谷歌自研TPU上运行,并便捷迁移到谷歌云。
谷歌称,开发者仅需简单修改配置,无需重写应用代码即可完成迁移。公司还推出TPU智能体套件,帮助开发者迁移应用并在过程中排查修复错误。
谷歌表示,其联合Meta、Hugging Face、Inferact、RadixArk等合作,保证TorchTPU与主流人工智能开发工具兼容;同时升级LiteRT工具,更好支持基于PyTorch开发的应用。
