jax

3a824c12 - Add is_device_rocm() check to CSR matmat and matvec test skip conditions (follow-up to #380)