diff --git a/pyproject.toml b/pyproject.toml
index f998a30..c5525b0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -79,6 +79,10 @@ torchvision = [
   { index = "pytorch-cu126", marker = "platform_system == 'Windows'" },
   { index = "pytorch-cu126", marker = "platform_system == 'Linux'" },
 ]
+triton = [
+  { index = "pytorch-cu126", marker = "platform_system == 'Windows'" },
+  { index = "pytorch-cu126", marker = "platform_system == 'Linux'" },
+]
 
 
 [[tool.uv.index]]