Add Python 3.12 support with torch 2.5.x

- Update Python version constraint from <3.12 to <3.13
- Upgrade torch from ^2.1.0 to ^2.5.0 (required for py3.12 + triton)
- Upgrade torchvision from ^0.16.0 to ^0.20.0
- Update CI workflows to test both Python 3.10 and 3.12

The triton package bundled with torch 2.1.x only had wheels for
Python 3.10/3.11. Torch 2.5.x includes triton 3.x which supports
Python 3.12.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: mihow

.github/workflows/test-ml.yml

@@ -15,13 +15,17 @@ permissions:
 jobs:
   build:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.12"]
+      fail-fast: false
 
     steps:
       - uses: actions/checkout@v3
-      - name: Set up Python 3.10
+      - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@main # Need latest version to use pyproject.toml instead of requirements.txt
         with:
-          python-version: "3.10"
+          python-version: ${{ matrix.python-version }}
           cache: "pip"
       - name: Install dependencies
         run: |

.github/workflows/test.yml

@@ -13,14 +13,18 @@ jobs:
   test:
     name: Run Python Tests
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.12"]
+      fail-fast: false
 
     steps:
       - uses: actions/checkout@v4
 
-      - name: Set up Python 3.10
+      - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: ${{ matrix.python-version }}
           cache: "pip"
           cache-dependency-path: |
             poetry.lock