Checkpointing your Machine Learning