Skip to content

Add separate mtp_grad_scale_func for MTP loss scaling - #3459

Merged
ericharper merged 2 commits into
NVIDIA:mainfrom
yfw:yifu/scale_mtp_loss
Jun 1, 2026
Merged

Add separate mtp_grad_scale_func for MTP loss scaling#3459
ericharper merged 2 commits into
NVIDIA:mainfrom
yfw:yifu/scale_mtp_loss

Commits

Commits on Jun 1, 2026