Skip to content

Add mesh rules that replicate dense weights across context (cp-as-ep-dense-no-cp, cp-as-ep-dense-fsdp) - #5783

Draft
parambole wants to merge 1 commit into
parambole/rl-trainer-4600from
parambole/ag-l2-dense-no-cp
Draft

Add mesh rules that replicate dense weights across context (cp-as-ep-dense-no-cp, cp-as-ep-dense-fsdp)#5783
parambole wants to merge 1 commit into
parambole/rl-trainer-4600from
parambole/ag-l2-dense-no-cp

Add cp-as-ep-dense-no-cp and cp-as-ep-dense-fsdp mesh rules

20c93dc
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Oct 15, 2026 in 8s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

✅ 20c93dc Author: @parambole <pa*****le​@google.com>

(Only the first commit for a unique contributor is listed.)

View more details on Google CLA