[Dev] replay: Cherry-pick: M4 + Dist Checkpoint: Replace global parallel state with explicit group parameters (#2152) + fix: m4 _clean_metadata_for_serialization for training - #2225