Qwen3 Embedding on Cloud TPU: Production Long-Context Retrieval with vLLM

Article Summary

Google Cloud published native vLLM TPU support for embedding inference on August 26, 2026, targeting production retrieval rather than chat generation. The engineering work focuses on Qwen3-Embedding-8B and Qwen3-VL-Embedding-8B with long text and multimodal contexts, including 16K-class text sequences and 15K+ multimodal inputs. Google addressed TPU tensor alignment, lazy loading, JAX/XLA compilation warm-up, chunked prefill, and pooling-state preservation through a hybrid StepPool design. In one published Qwen3-Embedding-8B configuration using bf16, 16K+ sequences, and TP=4, TPU Ironwood reached 83,996 total tokens/s and 5.13 requests/s. Google also validates cross-hardware vector parity with cosine-similarity thresholds of at least 0.999 for text and 0.995 for multimodal inputs.

Embedding infrastructure is easy to underestimate.

A prototype may look like:

documents