diff --git a/pyproject.toml b/pyproject.toml
index e60f6f3ff..bd184c49e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -63,3 +63,5 @@ docstring-code-format = false
 
 [tool.uv.extra-build-dependencies]
 axolotl = ["huggingface_hub"]
+flash-attn = [{ requirement = "torch", match-runtime = true }]
+deepspeed = [{ requirement = "torch", match-runtime = true }]