TPU 一直是 PyTorch 开发者眼中的「隔壁机房」——能看见,但进不去。这次 Google 在 PyTorch 相关公告中把门推开了:推出基于 TorchTPU 的 vLLM 和 SGLang 原生 PyTorch 推理引擎,开发者终于可以用自己熟悉的那套 PyTorch 工具链,把模型直接部署到 TPU 上。
重点在两个合作方。vLLM 版本与 Inferact 联手开发,SGLang 版本则交给 RadixArk。这两套框架如今几乎是大模型推理服务的事实标准,把它们的调度与批处理能力搬到 TPU 上,等于抽掉了「想用 vLLM 就得留在 GPU」这块默认前提。更实际的价值在于路径变短了:过去要上 TPU,往往得先转 JAX、再和 XLA 反复较劲,现在原生 PyTorch 这条路直接省掉了中间环节。
对 Google 而言,这是把 TPU 从训练场景的备选,往推理主力位置上推的一步棋。对开发者来说,意义不在于又多了一个后端选项,而在于算力选择权——同一份 PyTorch 代码,在 GPU 与 TPU 之间多了一条低成本的迁移通道。当然,算子覆盖、生态成熟度和真实吞吐还得看落地数据,但门槛确实已经降下来了。

