diff --git a/requirements.txt b/requirements.txt
index a4ec8da..d228129 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -51,7 +51,7 @@ distlib==0.3.9
     # via virtualenv
 fastapi==0.115.12
     # via -r requirements.in
-filelock==3.18.0
+filelock==3.20.1
     # via
     #   torch
     #   virtualenv
@@ -136,6 +136,45 @@ numpy==1.26.4
     #   torch-geometric
     #   torchvision
     #   xgboost
+nvidia-cublas-cu12==12.6.4.1
+    # via
+    #   nvidia-cudnn-cu12
+    #   nvidia-cusolver-cu12
+    #   torch
+nvidia-cuda-cupti-cu12==12.6.80
+    # via torch
+nvidia-cuda-nvrtc-cu12==12.6.77
+    # via torch
+nvidia-cuda-runtime-cu12==12.6.77
+    # via torch
+nvidia-cudnn-cu12==9.5.1.17
+    # via torch
+nvidia-cufft-cu12==11.3.0.4
+    # via torch
+nvidia-cufile-cu12==1.11.1.6
+    # via torch
+nvidia-curand-cu12==10.3.7.77
+    # via torch
+nvidia-cusolver-cu12==11.7.1.2
+    # via torch
+nvidia-cusparse-cu12==12.5.4.2
+    # via
+    #   nvidia-cusolver-cu12
+    #   torch
+nvidia-cusparselt-cu12==0.6.3
+    # via torch
+nvidia-nccl-cu12==2.26.2
+    # via
+    #   torch
+    #   xgboost
+nvidia-nvjitlink-cu12==12.6.85
+    # via
+    #   nvidia-cufft-cu12
+    #   nvidia-cusolver-cu12
+    #   nvidia-cusparse-cu12
+    #   torch
+nvidia-nvtx-cu12==12.6.77
+    # via torch
 onnx==1.17.0
     # via
     #   jaqpotpy
@@ -261,6 +300,8 @@ torchvision==0.22.0
     # via -r requirements.in
 tqdm==4.67.1
     # via torch-geometric
+triton==3.3.0
+    # via torch
 typing-extensions==4.13.2
     # via
     #   aiosignal