jax

dc51c478 - Merge pull request #34470 from ROCm:rocm_enable_tridiagonal_tests