jax

15c72e61 - Fix test_cuda_array_interface test skip condition (#657)