Merge branch 'add_automatic_checkpoint_and_restart' of...

Merge branch 'add_automatic_checkpoint_and_restart' of https://gitlab.jsc.fz-juelich.de/opengptx/bigscience-code into add_automatic_checkpoint_and_restart