Media Summary: This lighting talk provides a brief introduction to Lin Chai, Lance Wang, and Abheesht Sharma from Google talk about the latest features in the framework for post-training in Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Fall

Jax Devlab 2025 Jax Tunix - Detailed Analysis & Overview

This lighting talk provides a brief introduction to Lin Chai, Lance Wang, and Abheesht Sharma from Google talk about the latest features in the framework for post-training in Anxhelo Xhebraj introduces JaxPP, a library that enables MPMD training. Fall This tutorial provides an in-depth look at Do you want the speed and functional power of Jen Ha and Tom Yeh use spreadsheets to demonstrate the sharding in Qwen3. Fall

Try Brilliant free for 30 days You'll also get 20% off an annual premium subscription Kyle Meggs talks about the latest version of MaxText Fall Zeev Melumian talks about how to combine PyTorch and Minho Ryu, a Google Developer Expert in AI, talks about how you can contribute to the Unlock the full potential of your large language models with Matthew Johnson, Principal Scientist at Google, delivers the Keynote address at the Fall

David Hall gives the latest updates on Marin. Fall

Photo Gallery

JAX DevLab 2025 JAX: Tunix
Tunix: Tune in JAX | JAX/OpenXLA DevLab Fall 2025
JaxPP: A library for MPMD training in JAX | JAX/OpenXLA DevLab Fall 2025
JAX DevLab 2025 - Serving JAX Models
Unlocking Low-Level Control: Customizing Keras Training Loops with JAX
Models in JAX & NNX by Hand | JAX/OpenXLA DevLab Fall 2025
JAX in 100 Seconds
RL at Google Scale, with MaxText 2.0 | JAX/OpenXLA DevLab Fall 2025
TorchAX at Lightricks โ€” When PyTorch Meets JAX | JAX/OpenXLA DevLab Fall 2025
Contributing to the JAX Ecosystem | JAX/OpenXLA DevLab Fall 2025
How to fine-tune LLMs for with Tunix
Keynote address with Matt Johnson | JAX/OpenXLA DevLab Fall 2025
Sponsored
View Detailed Profile
JAX DevLab 2025 JAX: Tunix

JAX DevLab 2025 JAX: Tunix

This lighting talk provides a brief introduction to

Tunix: Tune in JAX | JAX/OpenXLA DevLab Fall 2025

Tunix: Tune in JAX | JAX/OpenXLA DevLab Fall 2025

Lin Chai, Lance Wang, and Abheesht Sharma from Google talk about the latest features in the framework for post-training in

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. Fall

JAX DevLab 2025 - Serving JAX Models

JAX DevLab 2025 - Serving JAX Models

This tutorial provides an in-depth look at

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

Sponsored
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. Fall

JAX in 100 Seconds

JAX in 100 Seconds

Try Brilliant free for 30 days https://brilliant.org/fireship You'll also get 20% off an annual premium subscription

RL at Google Scale, with MaxText 2.0 | JAX/OpenXLA DevLab Fall 2025

RL at Google Scale, with MaxText 2.0 | JAX/OpenXLA DevLab Fall 2025

Kyle Meggs talks about the latest version of MaxText Fall

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

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

How to fine-tune LLMs for with Tunix

How to fine-tune LLMs for with Tunix

Unlock the full potential of your large language models with

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 Fall

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. Fall