Support n_groups>1 for mamba2 #40861

tdoublep wants to merge 3 commits intohuggingface:mainhuggingface/transformers:mainfrom

tdoublep:mamba2-groupstdoublep/transformers:mamba2-groupsCopy head branch name to clipboard