Media Summary: Ying Chen and Harini Sridhar explain how to take advantage of Naums Mogers and Jevin Jiang talk about how compilation of Pallas kernels work for TPU TensorCores and Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the

Sparsecore Offloading Jax Openxla Devlab Fall 2025 - Detailed Analysis & Overview

Ying Chen and Harini Sridhar explain how to take advantage of Naums Mogers and Jevin Jiang talk about how compilation of Pallas kernels work for TPU TensorCores and Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Yu-Hang Tang from Nvidia talks about Post-training in Ihor Indyk introduces Grain, a Python library for data processing and loading for

Jiaxin Cao shows off how to execute async PjRT on GPU. Sharad Vikram explains how to use Pallas to make your custom kernels on TPUs go fast! Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the Zeev Melumian talks about how to combine PyTorch and

Photo Gallery

SparseCore Offloading | JAX/OpenXLA DevLab Fall 2025
Pallas TPU: Compilation for TensorCore & SparseCore | JAX/OpenXLA DevLab Fall 2025
Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025
JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025
Offloading RL Rollouts from JAX to vLLM for Efficient Post-Training | JAX/OpenXLA DevLab Fall 2025
JAX on GPUs | JAX/OpenXLA DevLab Fall 2025
Grain: Training data processing, tuning, and performance | JAX/OpenXLA DevLab Fall 2025
Asynchronous PjRT for JAX on GPU | JAX/OpenXLA DevLab Fall 2025
JAX/OpenXLA DevLab Fall 2025 Event Recap
Pallas TPU: New and Advanced Features for Kernels | JAX/OpenXLA DevLab Fall 2025
Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025
TorchAX at Lightricks โ€” When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025
Sponsored
View Detailed Profile
SparseCore Offloading | JAX/OpenXLA DevLab Fall 2025

SparseCore Offloading | JAX/OpenXLA DevLab Fall 2025

Ying Chen and Harini Sridhar explain how to take advantage of

Pallas TPU: Compilation for TensorCore & SparseCore | JAX/OpenXLA DevLab Fall 2025

Pallas TPU: Compilation for TensorCore & SparseCore | JAX/OpenXLA DevLab Fall 2025

Naums Mogers and Jevin Jiang talk about how compilation of Pallas kernels work for TPU TensorCores and

Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025

Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025

Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the

JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025

JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025

Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training.

Offloading RL Rollouts from JAX to vLLM for Efficient Post-Training | JAX/OpenXLA DevLab Fall 2025

Offloading RL Rollouts from JAX to vLLM for Efficient Post-Training | JAX/OpenXLA DevLab Fall 2025

Yu-Hang Tang from Nvidia talks about Post-training in

Sponsored
JAX on GPUs | JAX/OpenXLA DevLab Fall 2025

JAX on GPUs | JAX/OpenXLA DevLab Fall 2025

Kuy Mainwaring shows how to run

Grain: Training data processing, tuning, and performance | JAX/OpenXLA DevLab Fall 2025

Grain: Training data processing, tuning, and performance | JAX/OpenXLA DevLab Fall 2025

Ihor Indyk introduces Grain, a Python library for data processing and loading for

Asynchronous PjRT for JAX on GPU | JAX/OpenXLA DevLab Fall 2025

Asynchronous PjRT for JAX on GPU | JAX/OpenXLA DevLab Fall 2025

Jiaxin Cao shows off how to execute async PjRT on GPU.

JAX/OpenXLA DevLab Fall 2025 Event Recap

JAX/OpenXLA DevLab Fall 2025 Event Recap

Relive your favorite moments from the

Pallas TPU: New and Advanced Features for Kernels | JAX/OpenXLA DevLab Fall 2025

Pallas TPU: New and Advanced Features for Kernels | JAX/OpenXLA DevLab Fall 2025

Sharad Vikram explains how to use Pallas to make your custom kernels on TPUs go fast!

Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025

Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025

Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the

TorchAX at Lightricks โ€” When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025

TorchAX at Lightricks โ€” When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025

Zeev Melumian talks about how to combine PyTorch and

AMD's JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025

AMD's JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025

Andy Ye talks about how AMD is using