Bump minimum Jax requirement to 2.8.0 (#10027)

* Bump minimum Jax requirement to 2.8.0 * update table

Bump minimum Jax requirement to 2.8.0 (#10027)
* Bump minimum Jax requirement to 2.8.0 * update table
b9720dd6 · Patrick von Platen · GitHub · 89be094e · b9720dd6 · b9720dd6
Unverified Commit b9720dd6 authored Feb 05, 2021 by Patrick von Platen Committed by GitHub Feb 05, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

setup.py setup.py +1 -1

src/transformers/dependency_versions_table.py src/transformers/dependency_versions_table.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -102,7 +102,7 @@ _deps = [
    "importlib_metadata",
    "ipadic>=1.0.0,<2.0",
    "isort>=5.5.4",
-    "jax>=0.2.0",
+    "jax>=0.2.8",
    "jaxlib>=0.1.59",
    "keras2onnx",
    "numpy>=1.17",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -15,7 +15,7 @@ deps = {
    "importlib_metadata": "importlib_metadata",
    "ipadic": "ipadic>=1.0.0,<2.0",
    "isort": "isort>=5.5.4",
-    "jax": "jax>=0.2.0",
+    "jax": "jax>=0.2.8",
    "jaxlib": "jaxlib>=0.1.59",
    "keras2onnx": "keras2onnx",
    "numpy": "numpy>=1.17",