[DistMuon] Support oversharded QKV heads (#4078)