Marin An Open Laboratory For Foundation Models In Jax Jax Openxla Devlab Fall 2025
Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3. Sharad Vikram explains how to use Pallas to make your custom kernels on TPUs go fast! Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the Michael Goldfarb from Nvidia explains how to leverage CuTeDSL in
Matthew Watson, Fabien Hertschuh, and Abheesht Sharma explain how Keras can be used to run Zeev Melumian talks about how to combine PyTorch and Distinguished Engineer, Robert Hundt, from Google gives the keynote speech for the This tutorial provides an in-depth look at
Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025
Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3.
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 on GPUs | JAX/OpenXLA DevLab Fall 2025
Kuy Mainwaring shows how to run
JAX/OpenXLA DevLab Fall 2025 Event Recap
Relive your favorite moments from the
JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025
Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training.
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
Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025
Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the
Leveraging CuTeDSL in JAX | JAX/OpenXLA DevLab Fall 2025
Michael Goldfarb from Nvidia explains how to leverage CuTeDSL in
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
AMD's JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025
Andy Ye talks about how AMD is using
TorchAX at Lightricks — When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025
Zeev Melumian talks about how to combine PyTorch and
JAX/OpenXLA DevLab 2025 Keynote Speech with Robert Hundt
Distinguished Engineer, Robert Hundt, from Google gives the keynote speech for the
JAX DevLab 2025 - Serving JAX Models
This tutorial provides an in-depth look at
JAX-GCM: A differentiable atmospheric model | JAX/OpenXLA DevLab Fall 2025
Ellen Davenport introduces how to use