feat: refactor mcore train/forward utilities#1654

Merged

yuki-97 merged 19 commits intomainfrom

ashors/mcore-train

Feb 13, 2026

Commits on Feb 13, 2026

refactor megatron data utils

ashors1
authored and
ananthsub
committed
refactor train/forward utilities

ashors1
authored and
ananthsub
committed
add do_not_average_loss arg

ashors1
authored and
ananthsub
committed
remove unused import

ashors1
authored and
ananthsub
committed
add train unit tests

ashors1
authored and
ananthsub
committed
add unit tests

ashors1
authored and
ananthsub
committed
fix unit test

ashors1
authored and
ananthsub
committed
add missing copyright header

ashors1
authored and
ananthsub
committed
lint

ashors1
authored and
ananthsub
committed
lint

ashors1
authored and
ananthsub
committed
rebase & address feedback
ananthsub
committed
lint
ananthsub
committed
refactor
ananthsub
committed
address feedback
ananthsub
committed
update
ananthsub
committed
lint
ananthsub
committed
unit tests
ananthsub
committed
move metric aggregation to a function matching automodel
ananthsub
committed
Merge branch 'main' into ashors/mcore-train
yuki-97
authored