From notebook to production: Serving JAX at scale @googlecloudtech

Uploaded August 2026 | Updated September 2026, 2 weeks ago

Join the Google Cloud & NVIDIA community → https://g.dev/cloud/google-nvidia-community

Deploy models using robust JAX serving architectures. Watch along and learn how to achieve low latency and optimize setups specifically for web APIs.

* *Deploy AOT compilation:* Use ahead-of-time compilation to lock down input shapes and guarantee predictable inference latency.
* *Export native execution graphs:* Package model code and checkpoints using jax.export for portable runtime deployment.
* *Bridge JAX to TensorFlow serving:* Convert JAX graphs to standard TensorFlow SavedModels using jax2tf for corporate server integration.

This is part 4 of JAX on NVIDIA GPUs Crash Course.

Watch more JAX on NVIDIA GPUs Crash Course → https://g.dev/cloud/jax-nvidia-gpu
🔔 Subscribe to Google Cloud Tech → https://goo.gle/GoogleCloudTech

Speakers: Ivan Nardini, Ekaterina Sirazitdinova
Products Mentioned: Google Cloud, JAX, TensorFlow

From notebook to production: Serving JAX at scale

Build a calendar app in AI Studio in 6 minutes

$Make your website agent ready with WebMCP Join Gemini Enterprise Agent Ready (GEAR) for the latest agent resources here. → https://goo.gle/3Srblgl Access the source code on GitHub → https://goo.gle/4g8TC5t AI agents are becoming major consumers of the web, but todays websites are built for humans, not machines. WebMCP changes that. In this video, Hugo Zanini, a Google Developer Expert in AI, shows how to expose typed tools directly inside web pages so AI agents can act reliably, quickly, and at a fraction of the token cost. Watch along and learn: * The agent shopping problem and why current approaches like screenshot-scraping fall short. * How WebMCP works to expose contextual tools directly through simple browser APIs. * A live demo of Happy Coffee (a React-based developer portal) showing Gemini in action searching catalogs and analyzing quality checks. * WebMCP for Local Agents, demonstrating how local workflows can leverage web UIs to accelerate development. Watch more How I built my agent → https://www.youtube.com/playlist?list=PLIivdWyY5sqLseBL8iUmVbdd_h30C4oEo 🔔 Subscribe to Google Cloud Tech → https://goo.gle/GoogleCloudTech Speaker: Hugo Zanini Products Mentioned: WebMCP, React, Gemini$

Kickstart Conversational Analytics agents with the Looker ChromeUX Block

This AI generates video lessons in seconds | The Agent Factory

Unity Game Simulation: Find the perfect balance with Unity and GCP (Google Games Dev Summit)

Whats new in Google Clouds agent platform

How to design a multi-agent system that skips the LLM

How to speed up AI agents by 80% on the Gemini Enterprise Agent Platform