jax

979b537a - Correctly check dtype equality in `is_matmul_supported`.