Optimizing Flax NNX Models with Optax (Part 2)

From the creator

This is part two of our two episode series on optimizing models with Optax! This episode builds on the foundation of the first, exploring the powerful, composable features of Optax and showing how to scale training to multiple devices. 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.