跳到正文
vLLM Blog· Inferact and Google Teams·· 2 天前精选AI 评分67

vLLM 发布 TorchTPU 后端,支持在 Google Cloud TPU 上运行 vLLM

Introducing vLLM TorchTPU

AI 导读

vLLM 发布 vllm-torchtpu 后端,基于 Google 的 TorchTPU 栈在 Google Cloud TPU 上运行 vLLM,多数现有 vLLM 模型无需改动模型代码即可部署。

推荐理由

原文给出插件安装方式、Pallas 内核清单与六种并行维度,可据此判断在 TPU 上复用现有 vLLM 模型代码的迁移成本。

来源:vLLM Blog · vllm.ai