Skip to content

[PyTorch] Refactor grouped MLP into joint forward-backward fused op#3117

Merged
timmoon10 merged 2 commits into
NVIDIA:mainfrom
timmoon10:tmoon/refactor-grouped-mlp-op
Jun 11, 2026
Merged

[PyTorch] Refactor grouped MLP into joint forward-backward fused op#3117
timmoon10 merged 2 commits into
NVIDIA:mainfrom
timmoon10:tmoon/refactor-grouped-mlp-op

Review suggestions from @greptile-apps

6e79f15
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jun 10, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO