Enable jax 64bit

Author: JuliaPoo

.github/workflows/array-api-tests-jax.yml

@@ -7,3 +7,5 @@ jobs:
     uses: ./.github/workflows/array-api-tests.yml
     with:
       package-name: jax
+      extra-env-vars: |
+        JAX_ENABLE_X64=1