Thread MIMO support through the stock training loop (schedule + optimizer) - #5333

Merged

yashaswikarnati merged 4 commits into

NVIDIA:mainfrom

yashaswikarnati:ykarnati/upstream-trainloop-tl-bridge

Jun 16, 2026

Commits on Jun 15, 2026

training: thread p2p_communicator and schedule pg_collection into train loop

yashaswikarnati
and
claude
committed
8cbbd1b
optimizer: dispatch MimoModel to get_mimo_optimizer; thread pg_collection into setup

yashaswikarnati
and
claude
committed
95de128
Fix lint formatting for MIMO train loop PR
yashaswikarnati
committed
bef2ea1
test: synchronize dist checkpointing integrity file writes
yashaswikarnati
committed
065c920