智东西· 杨京丽·· 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