add github action

Author: jakevdp

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

@@ -0,0 +1,41 @@
+name: JAX Array API
+
+on:
+  workflow_dispatch: # allows triggering the workflow run manually
+  pull_request:  # Automatically trigger on pull requests affecting this file
+    branches:
+      - main
+    paths:
+      - '**workflows/jax-array-api.yml'
+      - '**experimental/array_api/**'
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: [3.11]
+
+    steps:
+    - name: Checkout jax
+      uses: actions/checkout@v3
+    - name: Checkout array-api-tests
+      uses: actions/checkout@v3
+      with:
+        repository: data-apis/array-api-tests
+        submodules: 'true'
+        path: 'array-api-tests'
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v1
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install .[cpu]
+        python -m pip install -r array-api-tests/requirements.txt
+    - name: Run the test suite
+      env:
+        ARRAY_API_TESTS_MODULE: jax.experimental.array_api
+      run: |
+        pytest --ci array-api-tests/array_api_tests/

pyproject.toml

@@ -57,6 +57,8 @@ markers = [
 ]
 filterwarnings = [
     "error",
+    "ignore:The numpy.array_api submodule is still experimental.:UserWarning",
+    "ignore:The hookimpl.*:DeprecationWarning",
     "ignore:No GPU/TPU found, falling back to CPU.:UserWarning",
     "ignore:xmap is an experimental feature and probably has bugs!",
     "ignore:the imp module is deprecated in favour of importlib.*:DeprecationWarning",