jax

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