Want to hear more about new vLLM and SGLang engines built on a new PyTorch-native TPU backend, TorchTPU?
Find Qi Zhou of
@Google, Colin Taylor and Angela Yi of
@Meta at PyTorch Conference North America in San Jose where they will discuss how they focused on making the existing
@sgl_project &
@vllm_project infrastructure work natively on TPU, preserving the schedulers, batching systems, OpenAI-compatible APIs, and torch.compile workflows already familiar to GPU users.
Because the serving engines remain upstream, new models and features can run on TPU without requiring separate TPU-specific implementations. Learn more about how it works across the stacks & what they learned running this in production, including compile time, device placement under tracing, KV cache layout, and cross-chip collectives.
Register for #
PyTorchCon# NA now: