Google 开源 TorchTPU,挑战 NVIDIA AI 生态主导地位

据PANews发布,PANews 10月21日消息,据 Google Cloud 官方博客,Google 在 PyTorch Conference 2026 上宣布开源 TorchTPU。该项目由 Google 与 Meta PyTorch 团队联合开发,代码已发布在 GitHub。TPU 由此成为与 GPU、CPU 并列的 PyTorch 原生硬件选项,现有代码只需少量修改即可迁移。vLLM、SGLang、TorchTitan、VeRL、Ray 等框架已经接入。在 Hugging Face Transformers v5 中,传入 device_map="tpu" 即可把模型部署到 TPU。
你怎么看这条消息?来投第一票

评论

0/500
免责声明:本文内容仅供参考,不构成任何投资建议。
更多快讯