[WS2][PR5][Logp] Add deterministic DP-aware GRPO loss on the TP-aware logprob path - #291

Open

KJLdefeated wants to merge 10 commits into

testfrom

feat/ws2-grpo-loss-pr5

Commits on Aug 2, 2026

feat(ws2): add TP-aware logprob contract and dispatch metadata
ryankert01
committed
b7ffb3a
fix(ws2): address CodeRabbit review on logprob contract PR
ryankert01
committed
cdc11ba
fix(ws2): address second CodeRabbit round on logprob contract
ryankert01
committed
6455715
feat(ws2): make determinism scope and invocation surface part of the typed contract
ryankert01
committed
3b4eaef
docs(ws2): drop standalone design doc per review
ryankert01
committed
e6dbeef
style(ws2): align comment density with sibling kernel modules
ryankert01
committed
878ba88

Commits on Aug 5, 2026

init vocab parallel logp
KJLdefeated
committed
a63bea2
init vocab parallel logp
KJLdefeated
committed
4fcdc30
adding cross tp testing
KJLdefeated
committed
6ffade4

Commits on Aug 12, 2026

ws2 deterministic grpo loss PR5
KJLdefeated
committed
3627fcd