Pin Jax to last working release (#16808)

* Pin Jax to last working release * Try lower * Try lower
2022-04-16 21:15:19 -04:00
parent 78f346c2b5
commit dee6f01636
2 changed files with 4 additions and 4 deletions
--- a/setup.py
+++ b/setup.py
@@ -113,8 +113,8 @@ _deps = [
    "importlib_metadata",
    "ipadic>=1.0.0,<2.0",
    "isort>=5.5.4",
-    "jax>=0.2.8,!=0.3.2",
-    "jaxlib>=0.1.65",
+    "jax>=0.2.8,!=0.3.2,<=0.3.6",
+    "jaxlib>=0.1.65,<=0.3.6",
    "jieba",
    "nltk",
    "numpy>=1.17",