build: Update lower bound on jax to v0.2.10 (#1666)

* Update the lower bound of the supported JAX versions to jax v0.2.10 and jaxlib v0.1.60 - jax v0.2.10 requires jaxlib v0.1.60+ * Set only lower bounds for jax and jaxlib in 'jax' extra
scikit-hep · Oct 23, 2021 · c3d0acb · c3d0acb
1 parent 6ade6d4
commit c3d0acb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lower-bound-requirements.txt b/lower-bound-requirements.txt
@@ -18,5 +18,5 @@ torch==1.10.0
 # Use Google Cloud Storage buckets for long term wheel support
 # c.f. https://github.com/google/jax/discussions/7608#discussioncomment-1269342
 --find-links https://storage.googleapis.com/jax-releases/jax_releases.html
-jax==0.2.8
-jaxlib==0.1.58
+jax==0.2.10
+jaxlib==0.1.60  # jax v0.2.10 requires jaxlib>=0.1.60
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
         'tensorflow-probability~=0.11',
     ],
     'torch': ['torch~=1.10'],
-    'jax': ['jax~=0.2.8', 'jaxlib~=0.1.58,!=0.1.68'],  # c.f. Issue 1501
+    'jax': ['jax>=0.2.10', 'jaxlib>=0.1.60,!=0.1.68'],  # c.f. Issue 1501
     'xmlio': ['uproot>=4.1.1'],
     'minuit': ['iminuit>=2.4'],
 }