diff --git a/pyproject.toml b/pyproject.toml
index 55c3ed69..38e131e6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,11 +23,11 @@ dependencies = [
     "catalogue>=2.0.10",
     "fabric>=3.2.2",
     "fiddle>=0.3.0",
-    "torchx>=0.7.0",
+    "torchx @ git+https://github.com/meta-pytorch/torchx",
     "typer>=0.12.3",
     "rich>=13.7.1",
     "jinja2>=3.1.4",
-    "cryptography < 43.0.0",
+    "cryptography",
     "networkx >= 3.3",
     "omegaconf>=2.3.0",
     "leptonai>=0.26.6",