jax

1179985d - [ROCm] Added a comment about why this test was enabled as per PR review.