Remove pinnings necessary on old jax versions

flferretti · Dec 6, 2023 · 458eeac · 458eeac
1 parent e0c0bc7
commit 458eeac
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -53,11 +53,10 @@ package_dir =
 python_requires = >=3.10
 install_requires =
     coloredlogs
-    jax >= 0.4.1, <0.4.11
-    jaxlib < 0.4.11
+    jax >= 0.4.1
+    jaxlib
     jaxlie
     jax_dataclasses >= 1.4.0
-    ml-dtypes < 0.3.0
     pptree
     rod
     typing_extensions; python_version < "3.11"