How to use custom JAX kernels with Pallas
3 699
16
Google Cloud Platform1.35 млн
Следующее
25.01.25 – 2 2550:23Популярные
14 дней – 6 5813:30
3 дня – 1 1244:01Опубликовано 25 января 2025, 17:00
Pallas quickstart → goo.gle/48bjFEN
Pallas design overview → goo.gle/408hpw4
Dive into the world of custom kernel programming in JAX with Pallas. Join Googler Paige Bailey as she provides an overview of Pallas, demonstrating how to define kernel functions, and leverage the power of Triton and Mosaic for efficient execution on GPUs and TPUs.
Watch more Generative AI Experiences for Developers → goo.gle/genAI4devs
Subscribe to Google Cloud Tech → goo.gle/GoogleCloudTech
Speaker: Paige Bailey
Products Mentioned: Pallas, JAX
#GoogleCloud #DevelopersAI
Pallas design overview → goo.gle/408hpw4
Dive into the world of custom kernel programming in JAX with Pallas. Join Googler Paige Bailey as she provides an overview of Pallas, demonstrating how to define kernel functions, and leverage the power of Triton and Mosaic for efficient execution on GPUs and TPUs.
Watch more Generative AI Experiences for Developers → goo.gle/genAI4devs
Subscribe to Google Cloud Tech → goo.gle/GoogleCloudTech
Speaker: Paige Bailey
Products Mentioned: Pallas, JAX
#GoogleCloud #DevelopersAI
Свежие видео
20 часов – 2020:23
5 дней – 11 74943:21
5 дней – 194 6368:21
Случайные видео
16 дней – 993 23720:50
65 дней – 1 8660:14
03.04.22 – 1 570 46213:43
14.06.21 – 84743:06
22.05.15 – 26 1533:41
3 дня – 47 86716:21
4 дня – 19 3090:32
5 дней – 2942:43
8 дней – 5 5971:13:27
10 дней – 16 5080:36
13 дней – 6 34310:06
13 часов – 3 3372:05