jax

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