diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
index 4e462095..a94bd1f9 100644
--- a/.github/workflows/pr-gpu.yaml
+++ b/.github/workflows/pr-gpu.yaml
@@ -21,14 +21,14 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "python3.11-pytorch2.6.0-gpus1"
+        - name: "python3.12-pytorch2.7.0-gpus1"
           gpu_num: 1
-          python_version: 3.11
-          container: mosaicml/pytorch:2.6.0_cu124-python3.11-ubuntu22.04
-        - name: "python3.11-pytorch2.6.0-gpus2"
+          python_version: 3.12
+          container: mosaicml/pytorch:2.7.0_cu128-python3.12-ubuntu22.04
+        - name: "python3.12-pytorch2.7.0-gpus2"
           gpu_num: 2
-          python_version: 3.11
-          container: mosaicml/pytorch:2.6.0_cu124-python3.11-ubuntu22.04
+          python_version: 3.12
+          container: mosaicml/pytorch:2.7.0_cu128-python3.12-ubuntu22.04
     steps:
     - name: Run PR GPU tests
       uses: mosaicml/ci-testing/.github/actions/pytest-gpu@v0.3.5
diff --git a/pyproject.toml b/pyproject.toml
index 54d4f5a2..d8471105 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@
 
 # build requirements
 [build-system]
-requires = ["setuptools < 79.0.0", "torch >= 2.6.0, < 2.6.1"]
+requires = ["setuptools < 79.0.0", "torch >= 2.7.0, < 2.7.1"]
 build-backend = "setuptools.build_meta"
 
 # Pytest
diff --git a/setup.py b/setup.py
index 1b80f17c..e625235e 100644
--- a/setup.py
+++ b/setup.py
@@ -62,15 +62,14 @@
 install_requires = [
     'numpy>=1.21.5,<2.1.0',
     'packaging>=21.3.0,<24.2',
-    'torch>=2.6.0,<2.6.1',
-    'triton>=3.2.0,<3.3.0',
+    'torch>=2.7.0,<2.7.1',
     'stanford-stk==0.7.1',
 ]
 
 extra_deps = {}
 
 extra_deps['gg'] = [
-    'grouped_gemm==0.2.0',
+    'grouped_gemm==0.3.0',
 ]
 
 extra_deps['dev'] = [