build: Move triton to optional

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

pyproject.toml

@@ -26,7 +26,6 @@ dependencies = [
 "accelerate>=0.20.3,!=0.34,<1.7",
 "transformers>=4.45,<4.52",
 "torch>=2.2.0,<2.6", 
-"triton>=3.0,<3.4",
 "tqdm>=4.66.2,<5.0",
 "datasets>=3.0.0,<4.0",
 "ninja>=1.11.1.1,<2.0",
@@ -47,6 +46,7 @@ mx = ["microxcaling>=1.1"]
 opt = ["fms-model-optimizer[fp8, gptq, mx]"]
 torchvision = ["torchvision>=0.17"]
 flash-attn = ["flash-attn>=2.5.3,<3.0"]
+triton = ["triton>=3.0,<3.4"]
 visualize = ["matplotlib", "graphviz", "pygraphviz"]
 dev = ["pre-commit>=3.0.4,<5.0"]
 test = ["pytest", "pillow"]