PR #4272 Porting RNN from Linen to NNX

Porting RNN from Linen to NNX #4272

copybara-service merged 5 commits into google:main from zinccat:rnn_nnx

added recurrent and test for lstmcell

8cd87ee1

fix seq_length in rnn, added test for rnn

d94b5a37

cgarciae commented on 2024-10-10

fix rng position, rng carry stream, scan arg

b19772eb

fix type

8d8073d9

Merge branch 'google:main' into rnn_nnx

8c86f8ec

IvyZX added pull ready

IvyZX approved these changes on 2024-10-16

copybara-service merged 3bf732c8 into main 1 year ago

Reviewers

IvyZX

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone