feat(loss): return per-depth MTP losses (#3525)
* feat(loss): return per-depth MTP losses Signed-off-by: SlyneD <slyned@nvidia.com> * fix(loss): preserve typed scalar MTP return Signed-off-by: SlyneD <slyned@nvidia.com> --------- Signed-off-by: SlyneD <slyned@nvidia.com>
S
Slyne Deng committed
71b2f869347a3110fd628859e9dfa5c2b49b90f8
Parent: a1de7b7
Committed by GitHub <noreply@github.com>
on 8/14/2026, 12:54:08 AM