Skip to content

Support efficient TEGroupedMLP (moe_grouped_gemm=True) for Minitron Pruning#1955

Draft
kevalmorabia97 wants to merge 1 commit into
mainfrom
kmorabia/minitron-moe-grouped-gemm
Draft

Support efficient TEGroupedMLP (moe_grouped_gemm=True) for Minitron Pruning#1955
kevalmorabia97 wants to merge 1 commit into
mainfrom
kmorabia/minitron-moe-grouped-gemm