diff --git a/pyproject.toml b/pyproject.toml
index 320155576..ec6edf107 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,3 +1,13 @@
 [tool.black]
 line-length = 100
-target-version = ['py38']
\ No newline at end of file
+target-version = ['py38']
+
+[build-system]
+requires = [
+    "torch",
+    "einops",
+    "packaging",
+    "ninja",
+    "wheel",
+    "setuptools"
+]