Videos#
A collection of videos related to the JAX AI Stack, including talks, tutorials, and presentations.
Learn more about TPUs, how they work, and how to use JAX and Pallas to use TPUs effectively in this video.
Watch this video for an overview of Megablocks, a library for building large-scale machine learning models with JAX.