Skip to content
Trending storyDeveloping

Google adds native PyTorch vLLM and SGLang backends for TPU, with vLLM TorchTPU release

2 articles2 sourcessince Oct 19Last article

Overview

AISummary of 2 articles

Google announces native PyTorch backends for the vLLM and SGLang inference engines on TPU, built on TorchTPU and developed with Inferact and RadixArk.

Google says the engines add prefix caching, KV cache offloading and disaggregated serving, with KV cache transfers handled by the open-source TPU Sync library, and that both already run at scale in production with preview customers.

vLLM separately releases vllm-torchtpu, a platform plugin that runs vLLM on Google Cloud TPUs without model code changes. It requires Python 3.12 on a Cloud TPU VM and installs a matching vLLM, torch, torch_tpu and libtpu. vLLM says TPU v7x (Ironwood) is the recommended hardware and is tested in CI on every pull request, while v6e, v5e and v5p are experimental.

Written by AI from the articles below · updated Oct 20, 12:55 PM ET

Check the sources:

Developments

2 developments

  1. Oct 19, 8:00 PM ET · 1 article
    Google launches native PyTorch vLLM and SGLang inference engines on TPU
    Google Developers Blog: Google launches native PyTorch vLLM and SGLang inference engines on TPU
  2. Oct 19, 8:00 PM ET · 1 article
    Inferact and Google release vLLM TorchTPU backend for Google Cloud TPUs
    vLLM Blog: vLLM releases TorchTPU backend for running vLLM on Google Cloud TPUs

Article timeline

The articles in this story. Times are ET.

Oct 19
  1. vLLM BlogOfficialPick
    vLLM releases TorchTPU backend for running vLLM on Google Cloud TPUs

    AIvLLM releases vllm-torchtpu, an out-of-tree platform plugin that runs vLLM on Google Cloud TPUs, so most models that already run in vLLM can be served on TPU without model code changes. The plugin keeps vLLM's CLI, OpenAI-compatible server, scheduler, and KV cache manager unchanged, and adds Pallas kernels for attention, Mixture-of-Experts, and quantized matmul.

  2. Google Developers BlogOfficialPick
    Google launches native PyTorch vLLM and SGLang inference engines on TPU

    AIGoogle announces native PyTorch backends for vLLM and SGLang on TPU, built on TorchTPU and developed with Inferact and RadixArk. The engines add production features such as prefix caching, KV cache offloading, and disaggregated serving, with KV cache transfers handled by the open-source TPU Sync library.

Heat Trend

Current comparable heat
12
Peak comparable heat
13Oct 20, 12:00 PM ET
Change in past 24 hours
–

The trend compares only continuously observed participants, so its scope may be smaller than the current heat total. Hover or tap for hourly heat; use arrow keys with a keyboard.