GitLab

* extending the current state_dict interface, make it possible to do everything in a single call, and to checkpoint on all ranks