diff --git a/README.md b/README.md
index 76b998ef..b50dbc4f 100644
--- a/README.md
+++ b/README.md
@@ -187,6 +187,13 @@ We also include an optimized reference implementation that uses [an optimized tr
 To install the reference Triton implementation run
 
 ```shell
+# You need to install triton from source to use the triton implementation
+git clone https://github.com/triton-lang/triton
+cd triton/
+pip install -r python/requirements.txt
+pip install -e . --verbose --no-build-isolation
+
+# Install the gpt-oss triton implementation
 pip install -e .[triton]
 ```
 
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",
 ]