[JAX] Bump min version (#25286)

* [JAX] Bump min version * make fixup
2023-08-03 16:05:02 +01:00
parent d114a6b71f
commit 66c240f3c9
2 changed files with 4 additions and 4 deletions
--- a/setup.py
+++ b/setup.py
@@ -124,8 +124,8 @@ _deps = [
    "importlib_metadata",
    "ipadic>=1.0.0,<2.0",
    "isort>=5.5.4",
-    "jax>=0.2.8,!=0.3.2,<=0.4.13",
-    "jaxlib>=0.1.65,<=0.4.13",
+    "jax>=0.4.1,<=0.4.13",
+    "jaxlib>=0.4.1,<=0.4.13",
    "jieba",
    "kenlm",
    "keras-nlp>=0.3.1",