jax

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