jax

bd6c64f6 - Fix test_cuda_array_interface test skip condition (#657)