fix: Unrestrict jaxlib upper bound and exclude jaxlib v0.1.68 (#1524)

* Unrestrict the upper bound on jaxlib in 'jax' extra - Reverts PR #1502 * Exclude jaxlib v0.1.68 from being installed as part of 'jax' extra - c.f. jax-ml/jax#7128
scikit-hep · Jul 10, 2021 · 025ffe9 · 025ffe9
1 parent 51a4a7f
commit 025ffe9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
         'tensorflow-probability~=0.10.1',
     ],
     'torch': ['torch~=1.8'],
-    'jax': ['jax~=0.2.8', 'jaxlib~=0.1.58,<0.1.68'],
+    'jax': ['jax~=0.2.8', 'jaxlib~=0.1.58,!=0.1.68'],  # c.f. Issue 1501
     'xmlio': [
         'uproot3>=3.14.1',
         'uproot~=4.0',