[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 = [
@@ -267,7 +279,7 @@ install_requires = [
 
 setup(
     name="transformers",
-    version="4.3.0.dev0", # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
+    version="4.3.0.dev0",  # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
     author="Thomas Wolf, Lysandre Debut, Victor Sanh, Julien Chaumond, Sam Shleifer, Patrick von Platen, Sylvain Gugger, Google AI Language Team Authors, Open AI team Authors, Facebook AI Authors, Carnegie Mellon University Authors",
     author_email="thomas@huggingface.co",
     description="State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch",

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