Skip to content

Add the missing b_hn bias to nnx.GRUCell - #5595

Open
vineethsaivs wants to merge 3 commits into
google:mainfrom
vineethsaivs:fix-nnx-grucell-hidden-bias-20260921
Open

vineethsaivs wants to merge 3 commits into
google:mainfrom
vineethsaivs:fix-nnx-grucell-hidden-bias-20260921

Broadcast and promote the GRUCell hn bias like Linear

998b0bf
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Oct 6, 2026 in 9s

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

✅ 998b0bf Author: @vineethsaivs <vine*******444​@gmail.com>

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