🚀 Feature

To add state_dict and load_state_dict to metrics to be able to save them to checkpoints and update them later.