AI, Data Science & Cloud Certificates from Google, IBM & Meta — 40% Off
One plan covers every Professional Certificate on Coursera. 40% off Coursera Plus Annual.
Unlock All Certificates
Learn advanced optimization techniques for Flax NNX models using Optax in this 12-minute tutorial that builds upon foundational concepts to explore powerful, composable features and demonstrate how to scale training across multiple devices.
