jax
Fix and enable RNN tests on JAX 0.5.0 branch
#301

Merged

Fix and enable RNN tests on JAX 0.5.0 branch #301

Ruturaj4 merged 1 commit into rocm-jaxlib-v0.5.0 from fix_rnn_5.0_branch

Fix and enable RNN tests on JAX 0.5.0 branch

9889427d

mrodden approved these changes on 2025-03-20

Ruturaj4 merged 41bcf9f1 into rocm-jaxlib-v0.5.0 1 year ago

Ruturaj4 deleted the fix_rnn_5.0_branch branch 1 year ago

Reviewers

mrodden

Assignees

No one assigned

Labels

None yet

Milestone

No milestone