Optimizing Flax NNX Models with Optax (Part 2)
Dec 4, 2025•Channel
AI Analysis
Data from YouTube Data API v3•Updated 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
Video Tags
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