JAX and Compiler Optimizations for Scale

JAX team describes customizing compiler/runtime to scale foundation models across tens to hundreds of thousands of GPUs.

Play episode from 24:20

Transcript

The AI-powered Podcast Player

Save insights by tapping your headphones, chat with episodes, discover the best highlights - and more!