jax

061408ac - Merge pull request #25803 from sergachev:fix_rnn_desc