From 8b8e94c632b67794a2769fc7d508aee01b10dab5 Mon Sep 17 00:00:00 2001
From: Vishal Krishna Kumar <vishal.myc2720@gmail.com>
Date: Sun, 28 Dec 2025 23:57:16 -0800
Subject: [PATCH] fix(install): add nvidia-nvshmem to GPU extras to prevent
 libnvshmem_host.so.3 error

---
 pyproject.toml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 3d03c4b..d3f5ce6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -66,6 +66,8 @@ cpu = [
 ]  
 gpu = [  
     "torch>=2.8.0",  
+    "nvidia-nvshmem-cu12>=3.3",
+
 ]  
   
 [tool.uv]