Scale AI with Google's TPU software stack

Unlock massive AI scale with a deep dive into Google's open-source software ecosystem. Explore high-performance tools designed to optimize the model lifecycle: pre-training with MaxText, post-training with Tunix, and inference with vLLM. Discover how to leverage these innovations alongside JAX and PyTorch, accelerated by infrastructure like Google TPUs, to build a state-of-the-art toolkit for next-generation AI.

Speakers: Josh Gordon, Girija Sathyamurthy, Rob Mulla

Watch the AI sessions from Google I/O 2026 → https://goo.gle/AI-at-IO26

Subscribe to Google for Developers → https://goo.gle/developers


#GoogleIO


Event: Google I/O 2026

Products Mentioned: AI/Machine Learning