The 2024 OpenXLA Spring Dev Lab in Sunnyvale!
See the event recap at OpenXLA Spring Dev Lab 2024: Building Groundbreaking ML Systems Together!
Date: April 25, 2024
Time: 8:30am-6:00pm
Location: 1195 Borregas Drive, Sunnyvale CA 94089 (US-SVL-MP6)
During this one-day event, you'll have the opportunity to select a specific XLA-related project and collaborate directly with an XLA engineer. This is a valuable chance to dive deep into OpenXLA development.
This event is primarily focused on hardware vendors, software infrastructure engineers, and compiler engineers who are actively contributing to the OpenXLA stack, or are looking to do so. We'll have OpenXLA Tutorials, roundtables, socials, and plenty of SWAG for attendees!
Schedule
Δ | Time | Content |
---|---|---|
30m | 8:30 - 9:00 | Networking Breakfast & Coffee |
25m | 9:00 - 9:25 | Intro & OpenXLA Overview |
1h | 9:30 - 10:30 | Tutorials - Round 1 |
1h | 10:40 - 11:40 | Tutorials - Round 2 |
1h 20m | 11:40 - 1:00 | Lunch, Social, Office Hours |
3h | 1:00 - 4:00 | Lab Sessions |
10m | 4:00 - 4:10 | Group Picture |
1h 40m | 4:10 - 6:00 | Happy Hour |
Tutorial Sessions
All tutorial sessions are held twice. Tutorials will be recorded and uploaded to the OpenXLA YouTube channel for those who can't make it!
Tutorial | Presenter |
---|---|
Extracting StableHLO from JAX, TensorFlow, and PyTorch (+Quantizing) | Kevin Gleason, Jen Ha & Xing Liu |
Integrating a Compiler & Runtime into PJRT | Jieying Luo & Skye Wanderman-Milne |
Optimizing Compute and Communication Scheduling For Your Hardware with XLA | TJ Xu, Abhinav Goel & Frederik Gossen |
Optimizing PyTorch Auto-sharding For Your Hardware | Yeounoh Chung & Pratik Fegade |
Discussion Tables
During Lab Sessions, tables will be set up specifically dedicated to discussion or questions on the following topics. Drop by at any time!
Discussion Topic | Host(s) |
---|---|
JAX | Peter Hawkins |
ML GPU Optimization for Large Scale Workloads | Tamás Danyluk |
PJRT | Xiao Yu, Aman Verma |
PyTorch XLA GPU | Milad Mohammadi |
StableHLO Quantizer | Andy Wan, Jen Ha, Xing Liu |
XLA PR Best Practices and Code Structure | Elliot English, Kyle Lucke |
See you there!
A few reminders leading up to the Dev Lab:
- Keep up to date on Dev Lab news at OpenXLA Discuss.
- Check you email for the latest registration details, including call for topics and tutorial registration.
- Connect with us on Discord for any additional questions!