NMFW-17: Add ColocatedBridgeCommunicator for heterogeneous TP/DP MIMO training#1

Closed

yashaswikarnati wants to merge 3 commits into

mainfrom

ykarnati/nmfw-17-colocated-colocated-bridge-communicator

Commits on Mar 28, 2026

Add ColocatedBridgeCommunicator for heterogeneous TP/DP MIMO training (NMFW-17)

yashaswikarnati
and
claude
committed
Add PP>1 support for LLM in colocated MIMO training (NMFW-19)

yashaswikarnati
and
claude
committed
Move correctness tests to separate PR (NMFW-50)

yashaswikarnati
and
claude
committed