Integrate GTP with heterogeneous MIMO training - #6268

Closed

yashaswikarnati wants to merge 8 commits into

NVIDIA:mainfrom

yashaswikarnati:ykarnati/mimo-gtp-stacked-e2e

Commits on Aug 4, 2026

Refine usage of pg_collection in gtp
fanshiqing
committed
4e87571
Pass gtp_remat process groups through pg_collection
fanshiqing
committed
2c47a2e
fix lint format
fanshiqing
committed
a6fffab
Fix gtp_remat resolution for partial pg_collections
fanshiqing
committed
7d26dc7
Pass process groups to remaining GTP layers
yashaswikarnati
committed
f6e5f84
Treat GTP ranks as bridge data lanes
yashaswikarnati
committed
10ff87a
Enable GTP in heterogeneous MIMO training
yashaswikarnati
committed
0732418
Use explicit GTP replica groups for checkpoints
yashaswikarnati
committed
edfd310