Squashed commit of the following:

commit 91851418a19d8015baeb13e1e48f1f763f75c2e0
Author: Julian Bilcke <[email protected]>
Date: Thu Aug 21 13:56:18 2025 +0200

small dependency fix

commit 46d0996f091f6f9ba102e9174cdc6d5f61b5a41c
Author: Julian Bilcke <[email protected]>
Date: Wed Aug 20 18:42:09 2025 +0200

small dependency fix

commit bcae8c0f0eea42f6a7d7d7a74d8a33b9e8bac297
Author: Julian Bilcke <[email protected]>
Date: Wed Aug 20 00:31:35 2025 +0200

small dependency fix

commit 5097c15de7df80c58f2ead1027ab9785b1cf2683
Author: Julian Bilcke <[email protected]>
Date: Tue Aug 19 19:48:26 2025 +0200

small dependency fix

commit 763997ce9e9cac0ebdddacb9d50f3c0625a1c1c6
Author: Julian Bilcke <[email protected]>
Date: Tue Aug 19 17:28:12 2025 +0200

fix python dependency issue

requirements.txt

@@ -39,9 +39,19 @@ networkx==3.3
 ninja==1.11.1.4
 numpy==2.1.2
 nvidia-ml-py==12.575.51
-nvidia-nccl-cu12==2.21.5
-nvidia-nvjitlink-cu12==12.4.127
-nvidia-nvtx-cu12==12.4.127
+
+# for Torch 2.8.0, we need to use the 2.27.3
+nvidia-nccl-cu12==2.27.3
+#nvidia-nccl-cu12==2.21.5
+
+# The user requested nvidia-nvjitlink-cu12==12.4.127
+# torch 2.8.0 depends on nvidia-nvjitlink-cu12==12.8.93
+nvidia-nvjitlink-cu12==12.8.93
+
+# The user requested nvidia-nvtx-cu12==12.4.127
+# torch 2.8.0 depends on nvidia-nvtx-cu12==12.8.90
+nvidia-nvtx-cu12==12.8.90
+
 nvitop==1.5.2
 opencv-python-headless==4.12.0.88
 packaging==25.0
@@ -59,11 +69,15 @@ safetensors==0.5.3
 sentencepiece==0.2.0
 setuptools==78.1.1
 six==1.17.0
-sympy==1.13.1
+sympy==1.13.3
 tokenizers==0.21.4
 tqdm==4.67.1
 transformers==4.54.1
-triton==3.1.0
+
+# The user requested triton==3.1.0
+# torch 2.8.0 depends on triton==3.4.0
+triton==3.4.0
+
 typing_extensions==4.12.2
 tzdata==2025.2
 urllib3==2.5.0