fix: Fixes to make Megatron backend match dtensor#1389

Merged

terrykong merged 19 commits intomainfrom

ashors/fix-mcore-scale

Oct 31, 2025

Commits on Oct 16, 2025

enable megatron calculate_per_token_loss
ashors1
committed

Commits on Oct 17, 2025

remove unneeded import
ashors1
committed

Commits on Oct 23, 2025

TP bug fix and add WIP unit test
ashors1
committed

Commits on Oct 24, 2025

fixes
ashors1
committed
lint
ashors1
committed

Commits on Oct 29, 2025

Commits on Oct 30, 2025

Commits on Oct 31, 2025