[Setup.py] update jaxlib (#9831)

* update jaxlib * Update setup.py * update table
2021-01-27 09:34:21 +01:00
parent f617490e71
commit d5b40d6693
2 changed files with 17 additions and 5 deletions
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -16,7 +16,7 @@ deps = {
    "ipadic": "ipadic>=1.0.0,<2.0",
    "isort": "isort>=5.5.4",
    "jax": "jax>=0.2.0",
-    "jaxlib": "jaxlib==0.1.55",
+    "jaxlib": "jaxlib>=0.1.59",
    "keras2onnx": "keras2onnx",
    "numpy": "numpy",
    "onnxconverter-common": "onnxconverter-common",