diff --git a/.github/workflows/build-ci.yml b/.github/workflows/build-ci.yml
index e2a6f0e..88c341b 100644
--- a/.github/workflows/build-ci.yml
+++ b/.github/workflows/build-ci.yml
@@ -18,7 +18,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v6
         with:
-          python-version: "3.10"
+          python-version: "3.13"
           cache: "pip"
           cache-dependency-path: pyproject.toml
 
@@ -37,7 +37,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v6
         with:
-          python-version: "3.10"
+          python-version: "3.13"
           cache: "pip"
           cache-dependency-path: pyproject.toml
 
@@ -54,30 +54,12 @@ jobs:
         run: |
           darglint src --strictness=short --ignore-raise=ValueError
 
-  tests-jax-latest:
+  tests:
     runs-on: ubuntu-latest
-    steps:
-      - name: Checkout repository
-        uses: actions/checkout@v5
-
-      - name: Set up Python
-        uses: actions/setup-python@v6
-        with:
-          python-version: "3.10"
-          cache: "pip"
-          cache-dependency-path: pyproject.toml
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.13"]
 
-      - name: Setup environment
-        run: |
-            python -m pip install --upgrade pip
-            pip install ".[tests,dev]"
-
-      - name: Run Python tests
-        run: |
-          pytest --cov=invrs_utils tests
-
-  tests-jax-0_4_36:
-    runs-on: ubuntu-latest
     steps:
       - name: Checkout repository
         uses: actions/checkout@v5
@@ -85,7 +67,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v6
         with:
-          python-version: "3.10"
+          python-version: ${{ matrix.python-version }}
           cache: "pip"
           cache-dependency-path: pyproject.toml
 
@@ -93,7 +75,6 @@ jobs:
         run: |
             python -m pip install --upgrade pip
             pip install ".[tests,dev]"
-            pip install --upgrade "jax==0.4.36"
 
       - name: Run Python tests
         run: |