jax

87e4b5f8 - Merge pull request #28036 from ROCm:ci_fix_rnn_lowering-upstream