jax

014cf308 - Merge pull request #26775 from ROCm:rocm-fix-numalib