Add CI for running a test on PRs

.github/workflows/run_tests.yml

@@ -0,0 +1,34 @@
+name: Run tests
+
+on:
+  pull_request:
+
+jobs:
+  run-test:
+    strategy:
+      matrix:
+        python-version: [ 3.10, 3.10, 3.11 ]
+        os: [ ubuntu-latest ]
+      fail-fast: false
+    runs-on: ${{ matrix.os }}
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install -r ./tests/requirements.txt
+
+      - name: Checks with pre-commit
+        uses: pre-commit/[email protected]
+
+      - name: Test with pytest
+        run: |
+          python -m pip install .
+          python -m pytest tests/

tests/requirements.txt

@@ -0,0 +1 @@
+jaxlib