diff --git a/.gitmodules b/.gitmodules
index d284208d0..d0d2cb3ef 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,3 +1,3 @@
 [submodule "csrc/sliding_tile_attention/tk"]
-	path = csrc/sliding_tile_attention/tk
+	path = sta_kernel/sliding_tile_attention/tk
 	url = https://github.com/HazyResearch/ThunderKittens.git
diff --git a/csrc/sliding_tile_attention/MANIFEST.in b/sta_kernel/sliding_tile_attention/MANIFEST.in
similarity index 100%
rename from csrc/sliding_tile_attention/MANIFEST.in
rename to sta_kernel/sliding_tile_attention/MANIFEST.in
diff --git a/csrc/sliding_tile_attention/README.md b/sta_kernel/sliding_tile_attention/README.md
similarity index 100%
rename from csrc/sliding_tile_attention/README.md
rename to sta_kernel/sliding_tile_attention/README.md
diff --git a/csrc/sliding_tile_attention/config.py b/sta_kernel/sliding_tile_attention/config.py
similarity index 100%
rename from csrc/sliding_tile_attention/config.py
rename to sta_kernel/sliding_tile_attention/config.py
diff --git a/csrc/sliding_tile_attention/setup.py b/sta_kernel/sliding_tile_attention/setup.py
similarity index 100%
rename from csrc/sliding_tile_attention/setup.py
rename to sta_kernel/sliding_tile_attention/setup.py
diff --git a/csrc/sliding_tile_attention/st_attn.cpp b/sta_kernel/sliding_tile_attention/st_attn.cpp
similarity index 100%
rename from csrc/sliding_tile_attention/st_attn.cpp
rename to sta_kernel/sliding_tile_attention/st_attn.cpp
diff --git a/csrc/sliding_tile_attention/st_attn/__init__.py b/sta_kernel/sliding_tile_attention/st_attn/__init__.py
similarity index 100%
rename from csrc/sliding_tile_attention/st_attn/__init__.py
rename to sta_kernel/sliding_tile_attention/st_attn/__init__.py
diff --git a/csrc/sliding_tile_attention/st_attn/st_attn_h100.cu b/sta_kernel/sliding_tile_attention/st_attn/st_attn_h100.cu
similarity index 100%
rename from csrc/sliding_tile_attention/st_attn/st_attn_h100.cu
rename to sta_kernel/sliding_tile_attention/st_attn/st_attn_h100.cu
diff --git a/csrc/sliding_tile_attention/test/bench.py b/sta_kernel/sliding_tile_attention/test/bench.py
similarity index 100%
rename from csrc/sliding_tile_attention/test/bench.py
rename to sta_kernel/sliding_tile_attention/test/bench.py
diff --git a/csrc/sliding_tile_attention/test/flex_sta_ref.py b/sta_kernel/sliding_tile_attention/test/flex_sta_ref.py
similarity index 100%
rename from csrc/sliding_tile_attention/test/flex_sta_ref.py
rename to sta_kernel/sliding_tile_attention/test/flex_sta_ref.py
diff --git a/csrc/sliding_tile_attention/test/test_sta.py b/sta_kernel/sliding_tile_attention/test/test_sta.py
similarity index 100%
rename from csrc/sliding_tile_attention/test/test_sta.py
rename to sta_kernel/sliding_tile_attention/test/test_sta.py
diff --git a/csrc/sliding_tile_attention/tk b/sta_kernel/sliding_tile_attention/tk
similarity index 100%
rename from csrc/sliding_tile_attention/tk
rename to sta_kernel/sliding_tile_attention/tk