jax

abbb2ef8 - Enabled RNN unit test: test_no_workspace_overflow for ROCm devices (#624)