From d5b40d669300b8deea0d0f98c83989e93ad38948 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Wed, 27 Jan 2021 09:34:21 +0100
Subject: [PATCH] [Setup.py] update jaxlib (#9831)

* update jaxlib

* Update setup.py

* update table
---
 setup.py                                      | 20 +++++++++++++++----
 src/transformers/dependency_versions_table.py |  2 +-
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index 126bf57b8a..bfa372cc5e 100644
--- a/setup.py
+++ b/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",
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index 69fc388f93..0f6fed0095 100644
--- 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",