25:28Grain: Training data processing, tuning, and performance | JAX/OpenXLA DevLab Fall 2025OpenXLA336 viewsView & Download
46:17Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab Fall 2025OpenXLA712 viewsView & Download
5:44JAX Data Loading: Using the Grain Dataset API for Simple and Declarative Data ProcessingGoogle for Developers2.4K viewsView & Download
25:17ML Performance Debugging with XProf | JAX/OpenXLA DevLab Fall 2025OpenXLA318 viewsView & Download
1:01:45JAX/OpenXLA DevLab 2025 - Large scale training techniques and best practicesOpenXLA1.0K viewsView & Download
44:47JAX/OpenXLA DevLab 2025 - Performance Debugging Deep Dive with XProfOpenXLA487 viewsView & Download
7:47Tokamax: A library of high-performance custom kernels | JAX/OpenXLA DevLab Fall 2025OpenXLA179 viewsView & Download
13:04Model Explorer: Visualization and Debugging for Large Model Graphs | JAX/OpenXLA DevLab Fall 2025OpenXLA282 viewsView & Download
21:55Accelerating LLM models on Blackwell | JAX/OpenXLA DevLab Fall 2025OpenXLA224 viewsView & Download
20:27Offloading RL Rollouts from JAX to vLLM for Efficient Post-Training | JAX/OpenXLA DevLab Fall 2025OpenXLA166 viewsView & Download
