jax

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