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 Eric Yang demonstrates the Model Explorer for visualizing and Lin Chai, Lance Wang, and Abheesht Sharma from Google talk about the latest features in the framework for post-training in Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Fall 2025 Devlab talks playlist: ... 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 ... Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3. Fall 2025 Devlab talks playlist: ... Distinguished Engineer, Robert Hundt, from Google gives the keynote speech for the Kuy Mainwaring shows how to run