2024 OpenXLA Dev Lab

OpenXLA Image

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!