fix: allow zero grad norm in dtensor policies for consistency with Megatron#1618
Closed
smahdavi4 wants to merge 4 commits intoNVIDIA-NeMo:mainfrom
Closed
fix: allow zero grad norm in dtensor policies for consistency with Megatron#1618smahdavi4 wants to merge 4 commits intoNVIDIA-NeMo:mainfrom
smahdavi4 wants to merge 4 commits intoNVIDIA-NeMo:mainfrom
Commits
Commits on Jan 13, 2026
- committed
- committed
Commits on Jan 14, 2026
- committed
- committed