CI: parametrize over Array API modules

Keep testing with array-api-strict, add numpy, torch and jax.numpy

Author: ev-br

.github/workflows/test.yml

@@ -8,7 +8,9 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.11"]
+        array-api-module: ["array_api_strict", "numpy", "torch", "jax.numpy"]
+      fail-fast: false
 
     steps:
     - name: Checkout array-api-tests
@@ -22,11 +24,18 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install array-api-strict
+        if [ ${{ matrix.array-api-module }} = 'torch' ]
+        then
+          python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
+        elif [ ${{ matrix.array-api-module }} = 'jax.numpy' ]
+          python -m pip install "jax[cpu]"
+        else
+          python -m pip install ${{ matrix.array-api-module }}
+        fi
         python -m pip install -r requirements.txt
     - name: Run the test suite
       env:
-        ARRAY_API_TESTS_MODULE: array_api_strict
+        ARRAY_API_TESTS_MODULE: ${{ matrix.array-api-module }}
         ARRAY_API_STRICT_API_VERSION: 2023.12
       run: |
         pytest -v -rxXfE --skips-file array-api-strict-skips.txt array_api_tests/