Google, IBM & Meta Certificates – 40% Off
One plan covers every Professional Certificate on Coursera.
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.
