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

Overview

Google, IBM & Meta Certificates — All 10,000+ Courses at 40% Off
One annual plan covers every course and certificate on Coursera. 40% off for a limited time.
Get Full Access
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

Reviews

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

Never Stop Learning.

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

Someone learning on their laptop while sitting on the floor.