Discover the cutting edge of foundation model development with JAX on Google Cloud. This session will showcase the latest advancements in the JAX ecosystem, including optimized performance on TPUs and GPUs. Explore new, high-performance models powered by MaxText and MaxDiffusion, delve into enhanced JAX libraries and Stable Stack packages, and learn about advanced diagnostics tools. Gain insights into how leading customers and partners are leveraging JAX on Google Cloud to build and deploy next-generation foundation models at scale.
talk-data.com
R
Speaker
Rajesh Anantharaman
2
talks
Senior Product Manager
Google Cloud
Filtering by:
Google Cloud Next '25
×
Filter by Event / Source
Talks & appearances
Showing 2 of 3 activities
with
Rajesh Anantharaman
(Google Cloud)
,
Ian Campbell
(Children's Hospital of Philadelphia)
,
Minho Ryu
(Kakao)
,
Nayeon Kim
(Kakao)
,
Kyle Meggs
(Google Cloud)
Maximize the efficiency of your machine learning workloads on Google Cloud infrastructure. This session delivers actionable strategies for in-depth diagnostics on both TPUs and GPUs. Learn to visualize model architectures, analyze critical performance metrics, and pinpoint optimization opportunities. Empower your data-driven decision-making and achieve significant performance gains through practical diagnostic techniques.