diff --git a/pyproject.toml b/pyproject.toml
index 48e9414..b0e3aa7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,9 +17,9 @@ authors = [
 
 requires-python = ">=3.12"
 dependencies = [
-  "chex~=0.1.88",
-  "jax~=0.4.38",
-  "tensorflow-probability[jax]~=0.25.0"
+  "chex>=0.1.88",
+  "jax>=0.4.38",
+  "tfp-nightly[jax]"
 ]
 
 [build-system]
@@ -28,5 +28,5 @@ build-backend = "setuptools.build_meta"
 
 [project.optional-dependencies]
 graph = [
-  "jraph~=0.0.6.dev0"
+  "jraph>=0.0.6.dev0"
 ]