diff --git a/python/pyproject_cpu.toml b/python/pyproject_cpu.toml
index 50406c14b3e6..257ac35eda65 100644
--- a/python/pyproject_cpu.toml
+++ b/python/pyproject_cpu.toml
@@ -100,6 +100,9 @@ dev = ["sglang[test]"]
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]
diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml
index cdc0fe1c2292..55007a25ee43 100755
--- a/python/pyproject_other.toml
+++ b/python/pyproject_other.toml
@@ -122,6 +122,9 @@ dev_hpu = ["sglang[all_hpu]", "sglang[test]"]
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]
diff --git a/python/pyproject_xpu.toml b/python/pyproject_xpu.toml
index 465f2b0f6964..3e88356dc5e7 100644
--- a/python/pyproject_xpu.toml
+++ b/python/pyproject_xpu.toml
@@ -104,6 +104,9 @@ dev = ["sglang[test]"]
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]