2024 OpenXLA Dev Lab

OpenXLA Image

Join us in Sunnyvale for the 2024 OpenXLA Dev Lab!

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!


Δ 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 Location
Extracting StableHLO from JAX, TensorFlow, and PyTorch (+Quantizing) Kevin Gleason, Jen Ha & Xing Liu TBD
Integrating a Compiler & Runtime into PJRT Jieying Luo & Skye Wanderman-Milne TBD
Optimizing Compute and Communication Scheduling For Your Hardware with XLA TJ Xu, Abhinav Goel & Frederik Gossen TBD
Optimizing PyTorch Auto-sharding For Your Hardware Yeounoh Chung & Pratik Fegade TBD

Recordings will be linked once available.

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!