jax

336e5872 - Merge pull request #34893 from magaonka-amd:rocm-fix-aot-test-platform