智东西· 杨 京丽·· 12 小时前精选AI 评分78
谷歌与 Meta 开源 TorchTPU,PyTorch 模型可迁移至 TPU
谷歌联手Meta奇袭英伟达,开源TorchTPU,给PyTorch代码搬家
AI 导读
谷歌宣布开源与 Meta PyTorch 团队共同开发的原生 TPU 软件栈 TorchTPU,开发者用较少代码改动即可把现有 PyTorch 模型部署到 TPU。
推荐理由
原文给出 TorchTPU 在训练与推理两侧的具体数字和工具接入范围,可据此判断 PyTorch 模型迁移 TPU 的实际门槛。
来源:智东西 · zhidx.com