diff --git a/setup.cfg b/setup.cfg
index 398189d00..d3aa7c991 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -53,10 +53,11 @@ package_dir =
 python_requires = >=3.10
 install_requires =
     coloredlogs
-    jax >= 0.4.1
-    jaxlib
+    jax >= 0.4.1, <0.4.11
+    jaxlib < 0.4.11
     jaxlie
     jax_dataclasses >= 1.4.0
+    ml-dtypes < 0.3.0
     pptree
     rod
     scipy