Getting started with JAX on NVIDIA GPUs @googlecloudtech
Getting started with JAX on NVIDIA GPUs  @googlecloudtech
Uploaded August 2026 | Updated September 2026, 2 weeks ago
Join the Google Cloud & NVIDIA community โ†’ https://g.dev/cloud/google-nvidia-community

Master the fundamentals of JAX on NVIDIA GPUs and learn how to configure a Google Kubernetes Engine (GKE) cluster for high performance GPU-accelerated computing. Join Ivan Nardini, AI Engineer at Google Cloud and Ekaterina Sirazitdinova, Developer Advocate at NVIDIA teach:
* *How to verify an NVIDIA GPU* visibility instantly using nvidia\-smi and JAX device diagnostics.
* *Discuss how the XLA compiler* traces and compiles Python code into optimized machine instructions.
* *How to eliminate performance bottlenecks* caused by host-to-device data transfers and asynchronous dispatch latency.

This is part 1 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
Getting started with JAX on NVIDIA GPUsWhat an AI agent built in 3 hours - Virtual try on for your wardrobeLet /goal take the wheel ๐Ÿ ๐ŸŽ๏ธ ๐Ÿ’จVideo avatar agent | The Agent Factory PodcastThe magic behind GenLatte: Flutter, Firebase, and GeminiMoving AI agents beyond โ€œHello Worldโ€ to real productionAdvanced RAG techniques for better retrieval performanceInside Google Antigravity 2.0: The complete developer guide | The Agent FactoryBeyond chatbots: How AI agents built a full OS, web apps, and moreAccelerate analytics and semantic search in real-time with AlloyDB for PostgreSQLFrom notebook to production: Serving JAX at scaleAutoscaling your AI agent under load
Google Cloud Tech |

Getting started with JAX on NVIDIA GPUs

SHARE TO X SHARE TO REDDIT SHARE TO FACEBOOK WALLPAPER