Add NAS to Minitron pruning for parameter based auto-pruning#720

Open

kevalmorabia97 wants to merge 9 commits intomainfrom

kmorabia/minitron-auto

Commits on Dec 23, 2025

Add parameter best auto-pruning for Minitron
kevalmorabia97
committed

Commits on Dec 30, 2025

Fix param count calculation + account depth pruning
kevalmorabia97
committed

Commits on Jan 2, 2026

Add score calculation logic
kevalmorabia97
committed

Commits on Jan 4, 2026

Commits on Jan 6, 2026

Fix dropping and reverting layers for mmlu eval
kevalmorabia97
committed

Commits on Jan 7, 2026

Allow skipping some hparams in NAS and further restrict search space
kevalmorabia97
committed

Commits on Jan 8, 2026