[Setup.py] update jaxlib (#9831)

* update jaxlib

* Update setup.py

* update table

setup.py

@@ -103,7 +103,7 @@ _deps = [
     "ipadic>=1.0.0,<2.0",
     "isort>=5.5.4",
     "jax>=0.2.0",
-    "jaxlib==0.1.55",
+    "jaxlib>=0.1.59",
     "keras2onnx",
     "numpy",
     "onnxconverter-common",
@@ -248,8 +248,20 @@ extras["dev"] = (
     + extras["modelcreation"]
 )
 
-extras["torchhub"] = deps_list("filelock", "importlib_metadata", "numpy", "packaging", "protobuf", "regex",
-                               "requests", "sacremoses", "sentencepiece", "torch", "tokenizers", "tqdm")
+extras["torchhub"] = deps_list(
+    "filelock",
+    "importlib_metadata",
+    "numpy",
+    "packaging",
+    "protobuf",
+    "regex",
+    "requests",
+    "sacremoses",
+    "sentencepiece",
+    "torch",
+    "tokenizers",
+    "tqdm",
+)
 
 # when modifying the following list, make sure to update src/transformers/dependency_versions_check.py
 install_requires = [

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",