2025 年 JAX 和 OpenXLA 夏季 DevLabs

2025 年 JAX 和 OpenXLA Summer DevLabs 将在 Sunnyvale 举办!

日期:2025 年 6 月 23 日和 24 日
时间:上午 8 点到下午 5 点
地点:加利福尼亚州森尼韦尔市

欢迎加入我们,参加即将在 Sunnyvale 举办的 2025 年 JAX 和 OpenXLA Summer DevLabs!在往年成功举办的基础上,我们将此次活动延长为两天。第一天将重点介绍 JAX,第二天将重点介绍 OpenXLA。您可以参加一天的活动,也可以参加两天的活动。

事件说明

本次活动将提供轻松的实践机会,让您了解 JAX 和 OpenXLA 开发,结识社区的其他成员,并与 Google 的工程师深入探讨具体需求。

时间表

JAX DevLab - 6 月 23 日(周一)

变动量 时间 内容
1 小时 上午 8:00 - 9:00 登记、简便早餐和咖啡
25 分钟 上午 9:00 - 9:25 主题演讲(Roy Frostig)
6500 万 上午 9:25 - 10:30 闪电秀。JAX 核心(Matthew Johnson)、Grain(Ihor Indyk)、Flax/NNX(Cristian Garcia)、Orbax(Colin Gaffney)、Tunix(Tianshu Bao)、Qwix(Dangyi Liu)、MaxText(Rich James)、分析和调试(Kelvin Le)、Pallas(Sharad Vikram)、Serving(Fenghui Zhang)。
20 分钟 上午 10:30 - 10:50 休息时间
60 分钟 11 点至 12 点 教程 1。模型开发和后期训练生态系统的新变化(Cristian Garcia、Tianshu Bao、Rich James)。分析工作负载(Kelvin Le)。大规模训练技巧和最佳实践 (Jacob Austin)。使用内核(Sharad Vikram)。
90 米 12:00 - 下午 1:30 午餐
60 分钟 下午 1:30 - 2:30 教程 #2。调试模型 (Skye Wanderman-Milne)。运行 JAX 模型(Brittany Rockwell、Pengchong Jin、Han Qi)。模型开发和后期训练生态系统的新变化(Cristian Garcia、Tianshu Bao、Rich James)。使用内核(Sharad Vikram)。
15 分钟 下午 2:30 - 2:45 休息时间
15 分钟 下午 2:45 - 3:00 合影和纪念品
90 米 下午 3:00 - 4:30 圆桌会
90 米 下午 4:30 - 6:00 欢聚时光

OpenXLA DevLab - 6 月 24 日(周二)

变动量 时间 内容
1 小时 上午 8:00 - 9:00 交流早餐会和咖啡会
20 分钟 上午 9:05 - 9:25 简短的主旨演讲
20 分钟 上午 9:25 - 9:45 休息时间
1 小时 上午 9:45 - 10:45 教程 #1 XLA 架构 (Blake Hechtman)。使用 Shardy 对工作负载进行分区(姜子轩)。PJRT (Parker Holloway)。XLA 中的运算符融合 (Karthik Srinivasa Murthy)。马赛克 (Jaswanth Sreeram)。PyTorch/XLA (韩琪)。
15 分钟 10:45 - 11m 休息时间
1 小时 11 点至 12 点 教程 #2:上午场次的重复,以便您可以参加两场。
1 小时 15 分钟 中午 12 点 - 下午 1 点 午餐
1 小时 下午 1:00 - 2:00 圆桌会议讨论
0 分钟 下午 2:00 合影和纪念品

我们会在活动结束后录制教程并将其上传到 OpenXLA YouTube 频道。我们会根据最终确定的时间表更新此页面。

登记方式

如果您想参加,请填写此表单以表达您的意向。空间有限,如果可以,我们会在 5 月 30 日之前确认您的报名。

有疑问?如有任何意见、我们期待听到您的反馈。您可以发送电子邮件至 devlab@google.com