跳到正文
智东西· 杨 京丽·· 12 小时前精选AI 评分78

谷歌与 Meta 开源 TorchTPU,PyTorch 模型可迁移至 TPU

谷歌联手Meta奇袭英伟达,开源TorchTPU,给PyTorch代码搬家

AI 导读

谷歌宣布开源与 Meta PyTorch 团队共同开发的原生 TPU 软件栈 TorchTPU,开发者用较少代码改动即可把现有 PyTorch 模型部署到 TPU。

推荐理由

原文给出 TorchTPU 在训练与推理两侧的具体数字和工具接入范围,可据此判断 PyTorch 模型迁移 TPU 的实际门槛。

来源:智东西 · zhidx.com