diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
index 594e74f40c0..64c00800171 100644
--- a/tensorflow/tools/pip_package/setup.py
+++ b/tensorflow/tools/pip_package/setup.py
@@ -54,13 +54,13 @@ _VERSION = '2.3.1'
 REQUIRED_PACKAGES = [
     'absl-py >= 0.7.0',
     'astunparse == 1.6.3',
-    'gast == 0.3.3',
+    'gast < 0.5',
     'google_pasta >= 0.1.8',
-    'h5py >= 2.10.0, < 2.11.0',
+    'h5py >= 2.10.0, < 4',
     'keras_preprocessing >= 1.1.1, < 1.2',
     # TODO(mihaimaruseac): numpy 1.19.0 has ABI breakage
     # https://github.com/numpy/numpy/pull/15355
-    'numpy >= 1.16.0, < 1.19.0',
+    'numpy < 2',
     'opt_einsum >= 2.3.2',
     'protobuf >= 3.9.2',
     'tensorboard >= 2.3.0, < 3',