diff --git a/python/pyproject.toml b/python/pyproject.toml
index 0d4c89240..dbe31e43f 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -72,7 +72,7 @@ dependencies = [
   "tiktoken",
   "tilelang==0.1.8",
   "timm==1.0.16",
-  "tokenspeed_mla==0.1.6",
+  "tokenspeed_mla==0.1.7",
   "torch==2.11.0",
   "torch_memory_saver>=0.0.9.post1",
   "torchao==0.17.0",