diff --git a/python/pyproject.toml b/python/pyproject.toml
index 9ee96f7392c..776213bcb5f 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -56,6 +56,7 @@ dependencies = [
   "sgl-kernel==0.3.13",
   "soundfile==0.13.1",
   "tiktoken",
+  "tilelang==0.1.6.post1",
   "timm==1.0.16",
   "torch==2.8.0",
   "torch_memory_saver==0.0.9rc1",