Commit 4d0f996

authored and

committed

Add MoE pruning support.

Signed-off-by: Keval Morabia <[email protected]>

1 parent 3a76d28 commit 4d0f996Copy full SHA for 4d0f996

2 files changed

+298

-32

lines changed

modelopt/torch
- nas/plugins
  - megatron.py
- prune/plugins
  - mcore_minitron.py

+298

-32

lines changed

Comments

(0)