diff --git a/.github/workflows/e2e-subtensor-tests.yaml b/.github/workflows/e2e-subtensor-tests.yaml
index b83308ad12..93e64e9826 100644
--- a/.github/workflows/e2e-subtensor-tests.yaml
+++ b/.github/workflows/e2e-subtensor-tests.yaml
@@ -205,7 +205,7 @@ jobs:
       - read-python-versions
     strategy:
       fail-fast: false
-      max-parallel: 16
+      max-parallel: 64
       matrix:
         include: ${{ fromJson(needs.find-tests.outputs.test-files) }}
     uses: ./.github/workflows/_run-e2e-single.yaml
diff --git a/.github/workflows/unit-and-integration-tests.yml b/.github/workflows/unit-and-integration-tests.yml
index 507a48c098..a6a525981e 100644
--- a/.github/workflows/unit-and-integration-tests.yml
+++ b/.github/workflows/unit-and-integration-tests.yml
@@ -54,7 +54,7 @@ jobs:
             uv-${{ runner.os }}-${{ runner.arch }}-py${{ matrix.python-version }}-
 
       - name: Sync dev deps (idempotent; fast on cache hit)
-        run: uv sync --extra dev --dev
+        run: uv sync --extra dev --dev --extra torch
 
       - name: Unit tests
         timeout-minutes: 20
diff --git a/pyproject.toml b/pyproject.toml
index 173aa491d6..91ec418fec 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -58,7 +58,6 @@ dev = [
     "aioresponses==0.7.6",
     "factory-boy==3.3.0",
     "types-requests",
-    "torch>=1.13.1,<3.0"
 ]
 torch = [
     "torch>=1.13.1,<3.0"