|
|
The 2025 JAX & OpenXLA Summer DevLabs in Sunnyvale!
Date: June 23rd and June 24th, 2025
Time: 8:00am-5:00pm
Location: Sunnyvale, CA
Join us for the upcoming 2025 JAX and OpenXLA Summer DevLabs in Sunnyvale! Building on the success of previous years, we've expanded this event to be two days. The first day will focus on JAX, and the second day will focus on OpenXLA. You're welcome to join us for one day, or for both.
Event Description
This event offers a casual, hands-on opportunity to learn about JAX and OpenXLA development, meet fellow members of the community, and deep dive into specific needs with engineers from Google.
Schedule
JAX DevLab - Monday 6/23
| Δ | Time | Content |
|---|---|---|
| 1h | 8:00 - 9:00am | Registration, light breakfast, and coffee |
| 25m | 9:00 - 9:25am | Keynote (Roy Frostig) |
| 65m | 9:25 - 10:30am | Lightning talks. JAX core (Matthew Johnson), Grain (Ihor Indyk), Flax/NNX (Cristian Garcia), Orbax (Colin Gaffney), Tunix (Tianshu Bao), Qwix (Dangyi Liu), MaxText (Rich James), Profiling and Debugging (Kelvin Le), Pallas (Sharad Vikram), Serving (Fenghui Zhang). |
| 20m | 10:30 - 10:50am | Break |
| 60m | 11 - 12pm | Tutorials #1. What's new in the model development and post-training ecosystem (Cristian Garcia, Tianshu Bao, Rich James). Profiling and optimizing workloads (Kelvin Le, Skye Wanderman-Milne). Large scale training techniques and best practices (Jacob Austin). Working with Kernels (Sharad Vikram). |
| 90m | 12:00 - 1:30pm | Lunch |
| 60m | 2:30 - 3:30pm | Tutorials #2. Repeat of the morning session, so you have a chance to attend two. Additional tutorial: Serving JAX Models (Fenghui Zhang). |
| 15m | 2:30 - 2:45pm | Break |
| 15m | 2:45 - 3:00pm | Group picture and swag |
| 90m | 3:00 - 4:30pm | Roundtables |
| 90m | 4:30 - 6:00pm | Happy hour |
OpenXLA DevLab - Tuesday 6/24
| Δ | Time | Content |
|---|---|---|
| 1h | 8:00 - 9am | Networking Breakfast & Coffee |
| 20m | 9:05 - 9:25am | Brief Keynote |
| 5m | 9:25 - 9:30am | Lightning talk (OSS Microbenchmarking, Chris Jones) |
| 1h | 9:30 - 10:30am | Tutorials #1 XLA Architecture (Blake Hechtman). Partitioning Workloads with Shardy (Zixuan Jiang). PJRT (Parker Holloway). Operator Fusion in XLA (Karthik Srinivasa Murthy). Mosaic (Jaswanth Sreeram). PyTorch/XLA (Han Qi). |
| 15m | 10:30 - 10:45am | Break |
| 1hr | 10:45 - 11:45m | Tutorials #2 Repeat of the morning sessions so you can attend two. |
| 1h 15m | 11:45 - 1:00pm | Lunch |
| 1h | 1:00 - 2:00pm | Roundtable discussions |
| 0m | 2:00pm | Group picture and swag |
Tutorials will be recorded and uploaded to the OpenXLA YouTube channel after the event. The schedule will be updated as it is finalized.
How to register
If you'd like to attend, please fill out this form to express interest. Space is limited. We will confirm your registration by 5/30 if we can!
Questions? Comments? We want to hear from you. You can email us at devlab@google.com