LLM

The Undo Button for AI: Building Resumable Training Loops with KerasHub and Orbax

Crane Hall – Workshop B

workshop
15:0040 mins13/11/2026

Every machine learning beginner has felt the heartbreak of losing hours of training progress because a browser-based cloud notebook disconnected, a laptop died, or a remote server environment timed out. This hands-on, practical workshop introduces the ultimate safety net for AI projects: resumable training loops.
Participants will learn how to load and fine-tune a model using KerasHub, and instantly pair it with Orbax—a powerful, open-source library built for robust model checkpointing. We will walk through the exact mechanics of creating a reliable “Save State” for your AI. By the end of this session, attendees will know how to smoothly save, restore, and resume their training mid-stream, transforming fragile code into resilient, production-ready ML workflows.