Leveraging the JAX AI Stack

From the creator

In this video we present an overview of the JAX AI Stack, focusing on how it can be used for high-performance model development and training. We'll draw parallels to PyTorch concepts, which you may be familiar with, to highlight the similarities and differences. Resources: Learn more → https://goo.gle/learning-jax Subscribe to Google for Developers → https://goo.gle/developers Speaker: Robert Crowe

Published 2025-12-04 · Watch on YouTube ↗
Availability and playback restrictions are controlled by the original channel.