jax

8cb89151 - Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors (#668)