Migrate tox.ini to tox.toml, add type checker compat matrix, streamline CI

- Replace tox.ini with modern tox.toml format using labels for env grouping
- Add type checker version compat envs: pyright (1.1.392+), mypy (1.15+), ty
- CI: run only default locked env (py312) on push; full compat + typecheck
  matrix triggers on version tags (v*)
- All tox envs use pytest-xdist (-n=auto via pytest.toml) for parallel tests

Author: acecchini

.github/workflows/ci.yml

@@ -3,13 +3,18 @@ name: CI
 on:
   push:
     branches: [main]
+    tags: ["v*"]
   pull_request:
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true
 
 jobs:
+  # -----------------------------------------------------------------------
+  # Lint — always run
+  # -----------------------------------------------------------------------
+
   ruff:
     runs-on: ubuntu-latest
     steps:
@@ -27,6 +32,10 @@ jobs:
       - uses: actions/checkout@v6
       - uses: codespell-project/actions-codespell@v2
 
+  # -----------------------------------------------------------------------
+  # Typecheck — always run
+  # -----------------------------------------------------------------------
+
   typecheck:
     runs-on: ubuntu-latest
     steps:
@@ -41,83 +50,97 @@ jobs:
       - uses: astral-sh/setup-uv@v7
       - run: uv run pytest tests/test_typecheck.py -v
 
+  # -----------------------------------------------------------------------
+  # Test — always run (single locked env, default Python)
+  # -----------------------------------------------------------------------
+
   test:
     needs: [ruff, codespell, typecheck]
     runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        python: ["3.12", "3.13", "3.14"]
-    name: test (py${{ matrix.python }})
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "py${{ matrix.python }}"
-        env:
-          TOX_PYTHON: python${{ matrix.python }}
+      - run: uv run tox run
 
-  numpy-compat:
-    needs: [ruff, codespell, typecheck]
-    runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        version: ["numpy22", "numpy23", "numpy24"]
-    name: ${{ matrix.version }}
-    steps:
-      - uses: actions/checkout@v6
-      - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "${{ matrix.version }}"
+  # -----------------------------------------------------------------------
+  # Full matrix — only on version tags (v*)
+  # -----------------------------------------------------------------------
 
-  jax-compat:
-    needs: [ruff, codespell, typecheck]
-    runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        version: ["jax05", "jax06", "jax07", "jax08", "jax09"]
-    name: ${{ matrix.version }}
-    steps:
-      - uses: actions/checkout@v6
-      - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "${{ matrix.version }}"
-
-  torch-compat:
-    needs: [ruff, codespell, typecheck]
+  test-matrix:
+    if: startsWith(github.ref, 'refs/tags/v')
+    needs: [test]
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
-        version: ["torch26", "torch27", "torch28", "torch29", "torch210"]
-    name: ${{ matrix.version }}
+        env: ["py313", "py314"]
+    name: ${{ matrix.env }}
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "${{ matrix.version }}"
+      - run: uv run tox run -e "${{ matrix.env }}"
 
-  beartype-compat:
-    needs: [ruff, codespell, typecheck]
+  compat:
+    if: startsWith(github.ref, 'refs/tags/v')
+    needs: [test]
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
-        version: ["bt020", "bt021", "bt022"]
-        suffix: ["", "-jax", "-torch"]
-    name: ${{ matrix.version }}${{ matrix.suffix }}
+        env:
+          - numpy22
+          - numpy23
+          - numpy24
+          - jax05
+          - jax06
+          - jax07
+          - jax08
+          - jax09
+          - torch26
+          - torch27
+          - torch28
+          - torch29
+          - torch210
+          - bt020
+          - bt021
+          - bt022
+          - bt020-jax
+          - bt021-jax
+          - bt022-jax
+          - bt020-torch
+          - bt021-torch
+          - bt022-torch
+          - optree014
+          - optree015
+          - optree016
+          - optree017
+          - optree018
+          - optree019
+    name: ${{ matrix.env }}
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "${{ matrix.version }}${{ matrix.suffix }}"
+      - run: uv run tox run -e "${{ matrix.env }}"
 
-  optree-compat:
-    needs: [ruff, codespell, typecheck]
+  typecheck-matrix:
+    if: startsWith(github.ref, 'refs/tags/v')
+    needs: [test]
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
-        version: ["optree014", "optree015", "optree016", "optree017", "optree018", "optree019"]
-    name: ${{ matrix.version }}
+        env:
+          - pyright1392
+          - pyright1400
+          - pyright1408
+          - mypy115
+          - mypy116
+          - mypy117
+          - mypy118
+          - mypy119
+          - ty
+    name: ${{ matrix.env }}
     steps:
       - uses: actions/checkout@v6
       - uses: astral-sh/setup-uv@v7
-      - run: uv run tox -e "${{ matrix.version }}"
+      - run: uv run tox run -e "${{ matrix.env }}"