智东西· 杨 京丽·· 14 小时前同新闻AI 评分61
谷歌与Meta开源TorchTPU,支持PyTorch模型在TPU上运行
谷歌联手Meta奇袭英伟达,开源TorchTPU,给PyTorch代码搬家
AI 导读
谷歌与Meta PyTorch团队共同开源TorchTPU,这是原生TPU软件栈,开发者可用较少代码改动将PyTorch模型部署到谷歌TPU上。TorchTPU已接入vLLM、SGLang、llm-d、TorchTitan、VeRL、Ray、Hugging Face和LiteRT等工具,覆盖大模型训练、推理、强化学习、集群调度及端侧部署场景。
同一新闻,精选展示《PyTorch 官方发布 TorchTPU:原生支持 Google TPU 的 PyTorch 后端》
来源:智东西 · zhidx.com