Skip to content

[DO NOT MERGE] perf(rl_trainer_bench): opt-in STAGE_BEFORE_UPDATE step boundary (-7.2% step time on the 397B bench; stacked on #5701) - #5719

Closed
parambole wants to merge 4 commits into
atwigg/mlperffrom
parambole/rl-bench-stage-before-update
Closed

[DO NOT MERGE] perf(rl_trainer_bench): opt-in STAGE_BEFORE_UPDATE step boundary (-7.2% step time on the 397B bench; stacked on #5701)#5719
parambole wants to merge 4 commits into
atwigg/mlperffrom
parambole/rl-bench-stage-before-update

rl_trainer_bench: STAGE_BEFORE_UPDATE switch for the step boundary

a03bcd4
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Oct 14, 2026 in 16s

✅ 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:

✅ a03bcd4 Author: @parambole <pa*****le​@google.com>
✅ 70d6444 Author: @ultrons <si*****av​@google.com>

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

View more details on Google CLA