feat(RL/post-2389): Phase 3+4 v2 client — compile-target registry + multi-source slice planner#349

Draft

KavinKrishnan wants to merge 40 commits into

mainfrom

kavink/post-2389-phase3-4

Commits on May 22, 2026

docs: add RL weight update integration slide deck

KavinKrishnan
authored and
jthomson04
committed
docs: add markdown version of RL integration slide deck

KavinKrishnan
authored and
jthomson04
committed
feat: add MxTrainingPublisher and MxRefitReceiver for RL weight refit

KavinKrishnan
authored and
jthomson04
committed
fix: list all sources and filter client-side by model_name (identity hash mismatch)

KavinKrishnan
authored and
jthomson04
committed
fix: register NIXL tensors only once per publisher lifetime

KavinKrishnan
authored and
jthomson04
committed
feat: add receive_weights_scratch() for cross-format RDMA transfers

KavinKrishnan
authored and
jthomson04
committed
fix: disable transfer coalescing in receive_weights_scratch (incompatible with scratch buffers)

KavinKrishnan
authored and
jthomson04
committed
fix: accept tensor_shapes in receive_weights_scratch for correct weight reshaping

KavinKrishnan
authored and
jthomson04
committed
chore: remove review/feedback docs from kavink/RL branch

KavinKrishnan
authored and
jthomson04
committed
docs: add MX RL integration overview (PRIME-RL + verl design)

KavinKrishnan
authored and
jthomson04
committed
docs: add component architecture diagrams for PRIME-RL and verl POCs

KavinKrishnan
authored and
jthomson04
committed
docs: add verl × ModelExpress integration overview with vertical diagrams

KavinKrishnan
authored and
jthomson04
committed
docs(verl): remove related-documents section with internal-only paths

KavinKrishnan
authored and
jthomson04
committed
docs(RL): cross-reference draft overlay PR #2343 in the design doc

KavinKrishnan
authored and
jthomson04
committed
docs(RL): refine VERL_MX_OVERVIEW for native nixl alignment and catalog value

KavinKrishnan
authored and
jthomson04
committed
docs(RL): add Path B native MX design as alternative to PI overlay

KavinKrishnan
authored and
jthomson04
committed
docs(RL): clarify v0.1 overlay scope in PRIMERL_MX diagrams

KavinKrishnan
authored and
jthomson04
committed
docs(RL): backfill PRIMERL_MX_OVERVIEW with Scenario A GB200 results

KavinKrishnan
authored and
jthomson04
committed
docs(RL): backfill PRIMERL_MX_OVERVIEW with B + C results

KavinKrishnan
authored and
jthomson04
committed
fix(RL): address CodeRabbit review on PR #252

KavinKrishnan
authored and
jthomson04
committed
docs(RL): add NIXL compression study reproduction guide

KavinKrishnan
authored and
jthomson04
committed
docs(RL): clarify NIXL compression data package is request-only

KavinKrishnan
authored and
jthomson04
committed
docs(RL): publish NIXL compression study capture scripts

KavinKrishnan
authored and
jthomson04
committed
docs(RL): genericize NIXL compression study + add component diagram

KavinKrishnan
authored and
jthomson04
committed
docs(RL): drop named NIXL inbox from compression study doc

KavinKrishnan
authored and
jthomson04
committed
feat(RL/NemoRL): v2 publisher/receiver with rank-to-rank, MoE expert filter, tree fan-out

KavinKrishnan
authored and
jthomson04
committed
feat(server): preserve SourceIdentity through GetMetadata for v2 RL clients

KavinKrishnan
authored and
jthomson04
committed
docs(RL): NemoRL × MX v2 overview — design, validation, deployment

KavinKrishnan
authored and
jthomson04
committed
docs(RL): scrub internal pensieve refs from NEMORL_MX_OVERVIEW

KavinKrishnan
authored and
jthomson04
committed
fix(RL/NemoRL): describe_tensor + sidecar carry shape_registry for DTensors

KavinKrishnan
authored and
jthomson04
committed
test(RL/NemoRL): add v2 DTensor E2E demo

KavinKrishnan
authored and
jthomson04
committed
docs(RL): NEMORL_MX_OVERVIEW — add §8.4 real-DTensor E2E results

KavinKrishnan
authored and
jthomson04
committed
fix(RL/NemoRL): drop v2 sidecar TensorDescriptors before NIXL register (#295)
jthomson04
committed
test(RL/NemoRL): FORCE_RDMA env var to exercise descriptor validation in loopback
KavinKrishnan
committed

Commits on May 23, 2026

fix(RL/NemoRL): post-rebase fixups for heartbeat path + receive_from_source signature
jthomson04
committed

Commits on May 27, 2026

feat(RL/post-2389): Phase 3+4 v2 client — compile-target registry + multi-source slice planner
KavinKrishnan
committed

Commits on May 28, 2026

feat(nemo_rl_v2): plumb compile_target + compile_metadata through add_tensor
KavinKrishnan
committed

Commits on May 30, 2026