Coursera Plus Annual is 40% Off — Ends July 6.

Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

Google

Checkpointing Flax NNX Models with Orbax - Part 2

Google via YouTube

Start learning Write review
Simplilearn Ad

Earn a Michigan Engineering AI Certificate — Stay Ahead of the AI Revolution

Learn More → Coursera Ad

AI, Data Science & Cloud Certificates from Google, IBM & Meta

Learn More →

Overview

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 to implement advanced checkpointing techniques for Flax NNX models using Orbax in this 12-minute tutorial that builds upon foundational concepts to cover optimizer state preservation and distributed model sharding strategies.

Syllabus

Checkpointing Flax NNX Models with Orbax (Part 2)

Taught by

Google Developers

Tags

Reviews

Start your review of Checkpointing Flax NNX Models with Orbax - Part 2

Start learning

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Sign up for free
Someone learning on their laptop while sitting on the floor.