Consider renaming `save_sampler_state` to `save_checkpoint`

[jzombie]

And there could be an additional load_checkpoint method which simply aliases to what's there now.

[jzombie]

Additionally, hooks could be provided so that these could also save/load model checkpoints when called.