Add training checkpoint (optimizer + scheduler state)

e042097

verified

freococo commited on 22 days ago