Skip to content

Fuse the Gated Delta Net inter-chunk scan into a Pallas TPU kernel#4348

Open
bzantium wants to merge 2 commits into
AI-Hypercomputer:mainfrom
bzantium:feat/gdn-pallas-scan
Open

Fuse the Gated Delta Net inter-chunk scan into a Pallas TPU kernel#4348
bzantium wants to merge 2 commits into
AI-Hypercomputer:mainfrom
bzantium:feat/gdn-pallas-scan

Force HIGHEST precision in the inversion VJP matmuls

63259a0
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jul 4, 2026 in 7s

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

✅ 63259a0 Author: @bzantium <ry****93​@gmail.com>

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

View more details on Google CLA