Save and Load a Checkpoint | AlgoML Prep