Add a 2.x guide to training checkpoints

A couple lines are lifted from the previous checkpoint guide, but it's mostly rewritten with the new APIs.

PiperOrigin-RevId: 228542666

Author: allenlavoie