jax

a83512ec - Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors

Skip test_tridiagonal_solve on ROCm due to hipSPARSE numerical errors hipsparseSgtsv2StridedBatch has precision issues with poorly-conditioned matrices. Related hipSPARSE ticket filed.