How to use custom JAX kernels with Pallas
3 428
15.4
Google Cloud Platform1.3 млн
Следующее
330 дней – 2 0310:23Популярные
60 дней – 1 8185:27
80 дней – 2 0174:40Опубликовано 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
Свежие видео
2 дня – 2 1570:16
3 дня – 2 3540:54
4 дня – 5 8725:00
92 дня – 1 016 9841:08Случайные видео
97 дней – 108 1350:14
182 дня – 171 6610:57
349 дней – 22 85527:56
04.07.23 – 177 59720:34
01.01.10 – 15 7231:40
14 дней – 79 50421:42
2 дня – 43 9934:18
3 дня – 8 7931:09:12
4 дня – 12 8560:27
10 дней – 830:22
10 дней – 116 64615:52
12 дней – 9 3401:00:36