Pinning of dependencies to support cu128 with arm64

Signed-off-by: Viktor Kuropiatnyk <vku@zurich.ibm.com>

pyproject.toml

@@ -98,8 +98,9 @@ cu126 = [
 ]
 
 cu128 = [
-  "torch>=2.7.1",
+  "torch~=2.7.1",
   "torchvision>=0.22.1",
+  "triton~=3.3.1 ; sys_platform == 'linux'"
 ]
 
 rocm = [
@@ -146,6 +147,10 @@ torchvision = [
   { index = "pytorch-rocm", group = "rocm", marker = "sys_platform == 'linux'" },
 ]
 
+triton = [
+  {index = "pytorch-cu128", group = "cu128", marker = "sys_platform == 'linux'" },
+]
+
 pytorch-triton-rocm = [
   { index = "pytorch-rocm", marker = "sys_platform == 'linux'" },
 ]