[CheckpointWrapper] Allow custom save and load functions through estimator tags

2 jobs for estimator-tags in 3 minutes and 57 seconds (queued for 1 second)