Media Summary: Michael Goldfarb from Nvidia explains how to leverage CuTeDSL in David Hall gives the latest updates on Marin. Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3.

Jaxpp A Library For Mpmd Training In Jax Jax Openxla Devlab Fall 2025 - Detailed Analysis & Overview

Michael Goldfarb from Nvidia explains how to leverage CuTeDSL in David Hall gives the latest updates on Marin. Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3. Matthew Watson, Fabien Hertschuh, and Abheesht Sharma explain how Keras can be used to run Sharad Vikram explains how to use Pallas to make your custom kernels on TPUs go fast! Distinguished Engineer, Robert Hundt, from Google gives the keynote speech for the

Zeev Melumian talks about how to combine PyTorch and Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the Jiaxin Cao shows off how to execute async PjRT on GPU. Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the Ying Chen and Harini Sridhar explain how to take advantage of SparseCores in XLA.

Photo Gallery

JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025
Leveraging CuTeDSL in JAX | JAX/OpenXLA DevLab Fall 2025
Marin: An Open Laboratory for Foundation Models in JAX | JAX/OpenXLA DevLab Fall 2025
Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025
Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab Fall 2025
Pallas TPU: New and Advanced Features for Kernels | JAX/OpenXLA DevLab Fall 2025
JAX/OpenXLA DevLab 2025 Keynote Speech with Robert Hundt
TorchAX at Lightricks β€” When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025
Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025
JAX/OpenXLA DevLab Fall 2025 Event Recap
Asynchronous PjRT for JAX on GPU | JAX/OpenXLA DevLab Fall 2025
Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025
Sponsored
View Detailed Profile
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

Leveraging CuTeDSL in JAX | JAX/OpenXLA DevLab Fall 2025

Leveraging CuTeDSL in JAX | JAX/OpenXLA DevLab Fall 2025

Michael Goldfarb from Nvidia explains how to leverage CuTeDSL in

Marin: An Open Laboratory for Foundation Models in JAX | JAX/OpenXLA DevLab Fall 2025

Marin: An Open Laboratory for Foundation Models in JAX | JAX/OpenXLA DevLab Fall 2025

David Hall gives the latest updates on Marin.

Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025

Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025

Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3.

Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab Fall 2025

Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab Fall 2025

Matthew Watson, Fabien Hertschuh, and Abheesht Sharma explain how Keras can be used to run

Sponsored
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!

JAX/OpenXLA DevLab 2025 Keynote Speech with Robert Hundt

JAX/OpenXLA DevLab 2025 Keynote Speech with Robert Hundt

Distinguished Engineer, Robert Hundt, from Google gives the keynote speech for 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

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

JAX/OpenXLA DevLab Fall 2025 Event Recap

JAX/OpenXLA DevLab Fall 2025 Event Recap

Relive your favorite moments from the

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.

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

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 SparseCores in XLA.