Skip to content

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

Merged
ksivaman merged 5 commits into
NVIDIA:mainfrom
ksivaman:relax_dim_constraint_for_fused_grouped_mlp
Apr 13, 2026
Merged

[PyTorch] Relax dimension constraints for using fused grouped MLP#2856
ksivaman merged 5 commits into
NVIDIA:mainfrom
ksivaman:relax_dim_constraint_for_fused_grouped_mlp

reset rng as before, assert input dim

f3d6950
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Apr 10, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO