谷歌与Meta PyTorch团队开源TorchTPU,可将PyTorch模型迁移至TPU
谷歌宣布开源与Meta PyTorch团队共同开发的原生TPU软件栈TorchTPU,开发者用较少代码改动即可把现有PyTorch模型部署到TPU上。该方案已接入vLLM、SGLang、llm-d、TorchTitan、VeRL、Ray、Hugging Face和LiteRT等工具,覆盖训练、推理、强化学习、集群调度和端侧部署。
推荐理由:谷歌开源原生TPU软件栈并给出8000颗Ironwood TPU的训练验证数据,可观察PyTorch生态向非英伟达硬件迁移的路径。