Skip to content

Unify Megatron and FSDP training interfaces with forward_backward + optim_step#901

Merged
erictang000 merged 9 commits intoNovaSky-AI:mainfrom
tyler-griggs:tgriggs/megatron_refactor
Jan 31, 2026
Merged

Unify Megatron and FSDP training interfaces with forward_backward + optim_step#901
erictang000 merged 9 commits intoNovaSky-AI:mainfrom
tyler-griggs:tgriggs/megatron_refactor

Commits

Commits on Jan 20, 2026

Commits on Jan 22, 2026

Commits on Jan 31, 2026