jax

f98aa2b2 - [ROCm]: Disable sparse_bcoo_bcsr_test.py::test_bcsr_dot_general for complex types