jax

2ce249ed - Merge pull request #34802 from magaonka-amd:enable-sparse-tests-rocm-upstream