Hands On Workshop Profiling Debugging Jax Openxla Workloads

Summary

Prasanna Rengasamy from Google and Chris Ashton from NVIDIA introduce how to profile and Kelvin Le walks through the features of XProf and shows a case study of how different hyperparameters affect performance. This tutorial gives you a tour of Lin Chai, Lance Wang, and Abheesht Sharma from Google talk about the latest features in the framework for post-training in Eric Yang demonstrates the Model Explorer for visualizing and Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3. Fall 2025 Devlab talks playlist: ... Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Fall 2025 Devlab talks playlist: ... Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the I started by talking about what XLA is actually doing, then looked at the HLO ops it generated for a Vision Transformer and their ... Ani Udipi introduces Tokamax, a library of high-performance custom kernels for TPU and GPU, so you can optimize your ML ... Distinguished Engineer, Robert Hundt, from Google gives the keynote speech for the