Google Cloud has introduced native vLLM support for Tensor Processing Units (TPUs) optimized for embedding inference, aiming for production retrieval systems. The update focuses on enhancing long-context and multimodal embeddings for models like Qwen3-Embedding-8B and Qwen3-VL-Embedding-8B, addressing challenges such as memory pressure and pooling correctness. Google's engineering efforts included optimizing tensor alignment, compilation warm-up, and a hybrid StepPool design to ensure mathematical correctness and high performance, with one configuration achieving over 83,000 tokens/s. AI
IMPACT Enables more efficient and accurate production-grade retrieval systems for long and multimodal documents.
RANK_REASON This is a significant infrastructure update for AI model serving, enabling new capabilities for long-context retrieval.
- Google Cloud
- Google Kubernetes Engine
- Jax
- Qwen3-Embedding-8B
- Qwen3-VL-Embedding-8B
- TPU Ironwood
- vLLM
- Xla
AI-generated summary · Google Gemini · from 2 sources. How we write summaries →