less strict bounds

Author: mattdangerw

requirements-jax-cuda.txt

@@ -1,13 +1,13 @@
-# Tensorflow (always required for preprocessing).
-tensorflow==2.14.0
-tensorflow-text==2.14.0.rc0
+# Tensorflow cpu-only version.
+tensorflow>=2.14.0
+tensorflow-text>=2.14.0.rc0
 
-# Torch cpu-only version (needed for testing).
+# Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.1
-torchvision==0.15.2
+torch>=2.0.1
+torchvision>=0.15.2
 
-# Jax.
+# Jax with cuda support.
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 jax[cuda11_pip]
 

requirements-tensorflow-cuda.txt

@@ -1,13 +1,13 @@
 # Tensorflow with cuda support.
-tensorflow[and-cuda]==2.14.0
-tensorflow-text[and-cuda]==2.14.0.rc0
+tensorflow[and-cuda]>=2.14.0
+tensorflow-text>=2.14.0.rc0
 
-# Torch cpu-only version (needed for testing).
+# Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.1
-torchvision==0.15.2
+torch>=2.0.1
+torchvision>=0.15.2
 
-# Jax cpu-only version (needed for testing).
+# Jax cpu-only version.
 jax[cpu]
 
 -r requirements-common.txt

requirements-torch-cuda.txt

@@ -1,13 +1,13 @@
-# Tensorflow (always required for preprocessing).
-tensorflow==2.14.0
-tensorflow-text==2.14.0.rc0
+# Tensorflow cpu-only version.
+tensorflow>=2.14.0
+tensorflow-text>=2.14.0.rc0
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu117
-torch==2.0.1+cu117
-torchvision==0.15.2+cu117
+torch>=2.0.1+cu117
+torchvision>=0.15.2+cu117
 
-# Jax cpu-only version (needed for testing).
+# Jax cpu-only version.
 jax[cpu]
 
 -r requirements-common.txt

requirements.txt

@@ -1,11 +1,11 @@
 # Tensorflow.
-tensorflow==2.14.0
-tensorflow-text==2.14.0.rc0
+tensorflow>=2.14.0
+tensorflow-text>=2.14.0.rc0
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.1
-torchvision==0.15.2
+torch>=2.0.1
+torchvision>=0.15.2
 
 # Jax.
 jax[cpu]