diff --git a/pyproject.toml b/pyproject.toml
index 08b0472..d6e57c7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,9 +3,9 @@ name = "needle"
 version = "0.1.0"
 requires-python = ">=3.11"
 dependencies = [
-    "jax",
-    "jaxlib",
-    "flax",
+    "jax>=0.4.30",
+    "jaxlib>=0.4.30",
+    "flax>=0.8.0",
     "optax",
     "datasets",
     "huggingface_hub",
diff --git a/requirements.txt b/requirements.txt
index eaf2d02..c361064 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
-jax
-jaxlib
-flax
+jax>=0.4.30
+jaxlib>=0.4.30
+flax>=0.8.0
 optax
 datasets
 huggingface_hub