Skip to content

397B v7x RL trainer: GDN kernel, MoE routing and engine speedups (10.62 -> 9.80 s per micro-batch on the 128-chip bench) - #5759

Closed
khatwanimohit wants to merge 11 commits into
atwigg/mlperffrom
mohit/trainer-perf-v7x
Closed

397B v7x RL trainer: GDN kernel, MoE routing and engine speedups (10.62 -> 9.80 s per micro-batch on the 128-chip bench)#5759
khatwanimohit wants to merge 11 commits into
atwigg/mlperffrom
mohit/trainer-perf-v7x

GDN CP backward: dS_ext_local as a reverse Pallas scan

4026f50
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:

✅ 4026f50 Author: @khatwanimohit <moh*******ani​@google.com>

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

View more details on Google CLA