Add support for TensorFlow 2.17 (#974)

Summary
- Update TensorFlow Quantum build to support TensorFlow 2.17.1.
- Align Eigen with TensorFlow 2.17.1’s pinned revision to fix
Tensor/Tensor.h build incompatibility.
- Define essential requirements in `requirements.in` and use
`pip-compile` to produce `requirements.txt`

Key changes
- WORKSPACE:
   - bump `@org_tensorflow` to 2.17.1
- add lightweight alias repo (pypi_setuptools) to satisfy TF’s
@pypi_setuptools reference when rules_python generates
@pypi__setuptools.
- `requirements.in`:
- set of requirements needed for building TFQ and running test scripts
- `requirements.txt` is produced by running `pip-compile --upgrade
--generate-hashes -o requirements.txt requirements.in`
 
Testing
- Linux/Ubuntu 22.04 (Docker), Python 3.11, Bazelisk → Bazel 6.5.0
- `bazel build ... release:build_pip_package` succeeds
- Unit tests (`scripts/test_all.sh` &
`./scripts/ci_validate_tutorials.sh`) pass
- Wheel installs and imports with TF 2.17.1 + tf-keras 2.17.0, Cirq
1.3.0, NumPy 1.26.4

Notes
- This keeps TFQ on Keras 2 path; Keras 3 migration can be handled in a
a future PR.
- This changes will help on the incoming upgrades to 2.20.

Author: psamanoelton

WORKSPACE

@@ -3,6 +3,11 @@
 
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
+local_repository(
+    name = "pypi_setuptools",
+    path = "third_party/pypi_setuptools",
+)
+
 # TensorFlow's .bzl files, loaded later in this file, also load rules_python
 # but we need a slightly newer version that is still compatible with TF's.
 http_archive(
@@ -39,11 +44,14 @@ load("@pypi//:requirements.bzl", "install_deps")
 
 install_deps()
 
-EIGEN_COMMIT = "aa6964bf3a34fd607837dd8123bc42465185c4f8"
+# Eigen commit used by TensorFlow / TFQ.
+# This commit corresponds to Eigen version 3.4.90
+# (verified via Eigen/src/Core/util/Macros.h).
+EIGEN_COMMIT = "c1d637433e3b3f9012b226c2c9125c494b470ae6"
 
 http_archive(
     name = "eigen",
-    sha256 = "35ba771e30c735a4215ed784d7e032086cf89fe6622dce4d793c45dd74373362",
+    sha256 = "0992b93a590c39e196a9efdb5b4919fbf3fb485e7e656c6a87b21ddadb7f6ad2",
     build_file_content = """
 cc_library(
   name = "eigen3",
@@ -71,10 +79,10 @@ http_archive(
 
 http_archive(
     name = "org_tensorflow",
-    patches = ["//third_party/tf:tf.patch"],
-    sha256 = "c8c8936e7b6156e669e08b3c388452bb973c1f41538149fce7ed4a4849c7a012",
-    strip_prefix = "tensorflow-2.16.2",
-    urls = ["https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.16.2.zip"],
+    # patches = ["//third_party/tf:tf.patch"],
+    sha256 = "75d63eab5c4b41a831e39e2e3795e1f9dcc247e8fcca11bf3f3e1a6ac35e55d0",
+    strip_prefix = "tensorflow-2.17.1",
+    urls = ["https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.17.1.zip"],
 )
 
 

release/setup.py

@@ -48,31 +48,29 @@ def finalize_options(self):
 REQUIRED_PACKAGES = [
     "cirq-core==1.3.0",
     "cirq-google==1.3.0",
-    "numpy<2.0",
-    "protobuf==4.25.8",
-    "scipy<=1.12.0",
     "sympy==1.14",
-    "tf-keras~=2.16.0",
-
+    "tf-keras~=2.17.0",
     # The following are transitive dependencies that need to be constrained to
     # avoid incompatible versions or because some (e.g., contourpy 1.3.3)
     # require Python 3.11+ and we want to maintain Python 3.9 compatibility.
     # TODO: revisit after we reach compatibility with TensorFlow 2.19+.
-    "contourpy<=1.3.0",
+    "contourpy<=1.3.2",
     "h5py==3.10.0",
     "importlib_metadata<5",
-    "jax<0.4.24",
-    "jaxlib<0.4.24",
+    "jax<=0.5",
     "matplotlib<3.10",
-    "networkx<3.3",
-    "pillow<=11.0",
+    "numpy<2.0",
+    "scipy<=1.12.0",
+    # The following makes it easier to get the right version on Colab. Once
+    # TFQ works with the latest version of TF, this may become unnecessary.
+    "protobuf==4.25.8",
 ]
 
 # TF requirement is placed as an extras to avoid overwriting existing nightly TF
 # installations. Users can run "pip install tensorflow-quantum[and-tensorflow]"
 # to get everything in one go (or "pip install tensorflow tensorflow-quantum").
 EXTRA_PACKAGES = {}
-EXTRA_PACKAGES["and-tensorflow"] = ["tensorflow>=2.16,<2.17"]
+EXTRA_PACKAGES["and-tensorflow"] = ["tensorflow>=2.17,<2.18"]
 # "extras" was used before 0.7.4. Prefer "and-tensorflow" in 0.7.4+.
 EXTRA_PACKAGES["extras"] = EXTRA_PACKAGES["and-tensorflow"]
 # Add an alias in case people type an underscore instead of a dash.
@@ -119,10 +117,10 @@ def has_ext_modules(self):
         "Intended Audience :: Education",
         "Intended Audience :: Science/Research",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
         "Programming Language :: Python :: 3.12",
+        "Programming Language :: Python :: 3.13",
         "Topic :: Scientific/Engineering",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
         "Topic :: Scientific/Engineering :: Mathematics",

requirements.in

@@ -18,8 +18,14 @@
 # scripts/generate_requirements.sh.
 cirq-core~=1.3.0
 cirq-google~=1.3.0
-tensorflow>=2.16,<2.17
-tf-keras~=2.16.0
+tensorflow>=2.17,<2.18
+keras==3.12.0
+tf-keras~=2.17.0
+setuptools==75.0.0
+wheel==0.44.0
+contourpy<=1.3.2
+networkx<=3.4.2
+scipy==1.15.3
 
 # TODO: the next ones are not truly core requirements. A better place should be
 # found for them (long with others needed by scripts/*). They're here as a