Media Summary: Matthew Watson, Fabien Hertschuh, and Abheesht Sharma explain how Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3.

Keras Deep Learning Framework For Jax Jax Openxla Devlab Fall 2025 - Detailed Analysis & Overview

Matthew Watson, Fabien Hertschuh, and Abheesht Sharma explain how Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3. Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the Do you want the speed and functional power of

Jiaxin Cao shows off how to execute async PjRT on GPU. Zeev Melumian talks about how to combine PyTorch and Michael Goldfarb from Nvidia explains how to leverage CuTeDSL in Cristian Garcia, creator of NNX, gives some key updates to the library.

Photo Gallery

Keras: Deep Learning Framework for JAX | JAX/OpenXLA DevLab Fall 2025
Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025
Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025
JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025
Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025
AMD's JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025
JAX on GPUs | JAX/OpenXLA DevLab Fall 2025
JAX/OpenXLA DevLab Fall 2025 Event Recap
Unlocking Low-Level Control: Customizing Keras Training Loops with JAX
Asynchronous PjRT for JAX on GPU | JAX/OpenXLA DevLab Fall 2025
TorchAX at Lightricks โ€” When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025
Leveraging CuTeDSL in JAX | JAX/OpenXLA DevLab Fall 2025
Sponsored
View Detailed Profile
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

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

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.

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.

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

Sponsored
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

JAX on GPUs | JAX/OpenXLA DevLab Fall 2025

JAX on GPUs | JAX/OpenXLA DevLab Fall 2025

Kuy Mainwaring shows how to run

JAX/OpenXLA DevLab Fall 2025 Event Recap

JAX/OpenXLA DevLab Fall 2025 Event Recap

Relive your favorite moments from the

Unlocking Low-Level Control: Customizing Keras Training Loops with JAX

Unlocking Low-Level Control: Customizing Keras Training Loops with JAX

Do you want the speed and functional power of

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.

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

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

Simplifying NNX: New features and updates | JAX/OpenXLA DevLab Fall 2025

Simplifying NNX: New features and updates | JAX/OpenXLA DevLab Fall 2025

Cristian Garcia, creator of NNX, gives some key updates to the library.