Oossa

Google adds native PyTorch support for vLLM and SGLang on TPU

Google launches PyTorch backends for the vLLM and SGLang inference engines, letting developers run these models on TPU hardware with familiar PyTorch tools.

NoteBy Published by Oossa: Last updated: 1 min read

Google announced new inference engines that run the open‑source vLLM and SGLang projects on TPUs using a native PyTorch backend called TorchTPU. The engines were co‑developed with Inferact (vLLM) and RadixArk (SGLang) and already serve production workloads at companies like Meta. They include TPU‑specific optimizations such as custom Pallas kernels and KV‑cache management, aiming to make large‑scale, low‑latency serving easier for PyTorch users.

Why it matters

PyTorch developers can now deploy fast, cost‑efficient inference for large language models on Google’s TPU hardware without switching to JAX.

Was this article useful?
Share

Read next

Oossa · Newsletter

The week in AI, explained

Every Monday: the stories worth knowing, in plain language. Free, no spam.