Optimizing Flax NNX Models with Optax (Part 2)

Dec 4, 2025Channel
AI Analysis
Data from YouTube Data API v3Updated Just now

Video Overview

Video Details

Published6 months ago
Duration12:02
Video IDzlySClpEIdc
Languageen
CategoryScience & Technology
PrivacyPublic
Made for KidsNo
Video TypeRegular Video

Performance Metrics

Views97
Likes4
Comments1
Engagement Rate5.15%
Likes per 100 views4.12
Comments per 1K views10.31

Description

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

Related Videos

More videos from Google for Developers