Add training state checkpointing#8

benbennett wants to merge 1 commit intomainbenbennett/SIMPLE:mainfrom

codex/find-method-to-save-training-statebenbennett/SIMPLE:codex/find-method-to-save-training-stateCopy head branch name to clipboard