diff --git a/pyproject.toml b/pyproject.toml
index 7289554..9fb0ae2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ opt = ["fms-model-optimizer[fp8, gptq, mx]"]
 aiu = ["ibm-fms>=0.0.8"]
 torchvision = ["torchvision>=0.17"]
 flash-attn = ["flash-attn>=2.5.3,<3.0"]
-triton = ["triton>=3.0,<3.5"]
+triton = ["triton>=3.0,<3.6"]
 visualize = ["matplotlib", "graphviz", "pygraphviz", "tensorboard", "notebook"]
 dev = ["pre-commit>=3.0.4,<5.0"]
 test = ["pytest", "pillow"]