[PyTorch] Relax dimension constraints for using fused grouped MLP#2856

ksivaman wants to merge 4 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

ksivaman:relax_dim_constraint_for_fused_grouped_mlpksivaman/TransformerEngine-1:relax_dim_constraint_for_fused_grouped_mlpCopy head branch name to clipboard