Support the MIMO cross-grid path in training loop - #5373

Merged

yashaswikarnati merged 5 commits into

NVIDIA:mainfrom

yashaswikarnati:p5-training-guards

Jun 22, 2026

Commits on Jun 16, 2026

training: guard MIMO cross-grid edge cases in the stock train loop
yashaswikarnati
committed
b4a3b7f

Commits on Jun 17, 2026

training: trim verbose MIMO guard comments

yashaswikarnati
and
claude
committed
71c1337
training: gate modelopt distillation shape-adjust on modelopt_enabled

yashaswikarnati
and
claude
committed
5d4f2ad
training: drop schedule-selection comment in MIMO branch

yashaswikarnati
and
claude
committed
30a54c7
training: select bridge schedule via get_forward_backward_func

yashaswikarnati
and
claude
committed
649065d