vLLM 与 Google 合作推出 vLLM-TorchTPU,支持 TPU 推理
vLLM 和 Google 一起搞了个 vLLM-TorchTPU,用 PyTorch 就能在 TPU 上跑推理,还带前缀缓存和投机解码,不想绑死 GPU 的可以看看。
vLLM 团队与 Google、inferact 合作发布了 vLLM-TorchTPU,让开发者可以在 TPU 上用原生 PyTorch 工具做模型推理。该方案支持前缀缓存、投机解码和分离式部署等功能。这延续了 vLLM 跨硬件支持 P10、TPU 等多类推理芯片的策略。
vLLM 🤝 TPU
🚀 Native PyTorch serving on TPUs is here! Built with @Google and @inferact, vLLM-TorchTPU brings familiar PyTorch tools to TPU inference, with prefix caching, speculative decoding, and disaggregated serving.
🛠️ Excited to keep building with the community!