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

谷歌开源TorchTPU,助力PyTorch代码直接迁移至TPU

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

AI 导读

谷歌与Meta PyTorch团队联合开源原生TPU软件栈TorchTPU,开发者可通过少量代码改动将PyTorch模型部署至TPU,并在最多8000颗Ironwood TPU上完成训练验证,Qwen3-8B测试浮点运算利用率最高达52%。

推荐理由

开源软件栈打通了PyTorch在TPU上的完整链路,读者能据此迁移模型并降低对单一硬件的依赖。

来源:智东西 · mp.weixin.qq.com