diff --git a/.github/workflows/python-lints.yml b/.github/workflows/python-lints.yml
index 9b708a3..4b7b487 100644
--- a/.github/workflows/python-lints.yml
+++ b/.github/workflows/python-lints.yml
@@ -8,7 +8,7 @@ jobs:
     strategy:
       max-parallel: 4
       matrix:
-        python-version: [3.9, 3.10, 3.11]
+        python-version: [3.9, "3.10", 3.11]
 
     steps:
     - uses: actions/checkout@v1
diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml
index 354f44a..d2f181f 100644
--- a/.github/workflows/python-tests.yml
+++ b/.github/workflows/python-tests.yml
@@ -8,12 +8,13 @@ jobs:
     strategy:
       max-parallel: 4
       matrix:
-        - python-version: 3.9
-          pt-version: 1.10.2
-        - python-version: "3.10"
-          pt-version: 2.1.2
-        - python-version: "3.11"
-          pt-version: 2.4.1
+        include:
+          - python-version: 3.9
+            pt-version: 1.10.2
+          - python-version: "3.10"
+            pt-version: 2.1.2
+          - python-version: "3.11"
+            pt-version: 2.4.1
 
     steps:
     - uses: actions/checkout@v1
diff --git a/.gitignore b/.gitignore
index 046fb96..f0efd4c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -10,7 +10,6 @@ __pycache__
 build/
 dist/
 
-.github/
 
 # Test files
 logs/*
diff --git a/setup.cfg b/setup.cfg
index 8d5c114..3907885 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -19,6 +19,9 @@ deps =
     pt1.13.1: torch==1.13.1+cu117
     pt1.13.1: numpy<2.0
     pt2.1.2: torch==2.1.2+cu121
+    pt2.1.2: numpy<2.0
+    pt2.4.1: torch==2.4.1
+    pt2.4.1: numpy<2.0
 install_command = pip install --find-links https://download.pytorch.org/whl/torch_stable.html {opts} {packages}
 commands =
     pytest tests