diff --git a/pyproject.toml b/pyproject.toml
index f3cd8d7..944a29d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -63,6 +63,7 @@ cpu = [
 ]
 gpu = [
     "torch>=2.9.1",
+    "nvidia-nvshmem-cu12>=3.3",
 ]
 
 [tool.uv]