diff --git a/pyproject.toml b/pyproject.toml
index 72d7411e..f00c72df 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,8 +24,7 @@ version = "0.0.1"
 
 [project.optional-dependencies]
 triton = [
-  "triton @ git+https://github.com/triton-lang/triton.git",
-  "triton_kernels @ git+https://github.com/triton-lang/triton.git#subdirectory=python/triton_kernels",
+  "triton",
   "safetensors>=0.5.3",
   "torch>=2.7.0",
 ]