diff --git a/.bazelrc b/.bazelrc
index fd7cb10632..e2754d2340 100644
--- a/.bazelrc
+++ b/.bazelrc
@@ -149,4 +149,4 @@ build --cxxopt=-D_GLIBCXX_USE_CXX11_ABI=1
 # Default startup
 startup --host_jvm_args=-Dbazel.DigestFunction=SHA256
 
-build --define=build_with_onednn_v2=true
+build --copt=-DDENABLE_ONEDNN_V3
diff --git a/BUILD.bazel b/BUILD.bazel
new file mode 100644
index 0000000000..3b35b77c9e
--- /dev/null
+++ b/BUILD.bazel
@@ -0,0 +1,34 @@
+load("@python//:defs.bzl", "compile_pip_requirements")
+load("@python_version_repo//:py_version.bzl", "REQUIREMENTS")
+load("@rules_license//rules:license.bzl", "license")  # copybara:strip
+
+license(
+    name = "package_license",
+    package_name = "tensorflow_federated",
+)
+
+licenses(["notice"])
+
+exports_files([
+    "copy.bara.sky",
+    "LICENSE",
+    "README.md",
+])
+
+filegroup(
+    name = "pyproject_toml",
+    srcs = ["pyproject.toml"],
+    visibility = ["//tensorflow_federated/tools/python_package:python_package_tool"],
+)
+
+compile_pip_requirements(
+    name = "requirements",
+    extra_args = [
+        "--allow-unsafe",
+        "--build-isolation",
+        "--rebuild",
+    ],
+    generate_hashes = True,
+    requirements_in = "requirements.in",
+    requirements_txt = REQUIREMENTS,
+)
diff --git a/RELEASE.md b/RELEASE.md
index 364f7a0e72..4cc9f9374e 100644
--- a/RELEASE.md
+++ b/RELEASE.md
@@ -25,6 +25,7 @@ and this project adheres to
 *   `DPGroupByFactory::CreateInternal`: when `delta` is not provided, check if
     the right norm bounds are provided to compute L1 sensitivity (for the
     Laplace mech).
+*   `tensorflow` version increased to 2.16.1.
 
 ## Release 0.84.0
 
diff --git a/WORKSPACE b/WORKSPACE
index 541bb47dfa..d6909d8db3 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -18,64 +18,75 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 http_archive(
     name = "bazel_skylib",
-    url = "https://github.com/bazelbuild/bazel-skylib/archive/refs/tags/1.3.0.tar.gz",
     sha256 = "3b620033ca48fcd6f5ef2ac85e0f6ec5639605fa2f627968490e52fc91a9932f",
     strip_prefix = "bazel-skylib-1.3.0",
+    url = "https://github.com/bazelbuild/bazel-skylib/archive/refs/tags/1.3.0.tar.gz",
 )
 
 http_archive(
     name = "com_github_grpc_grpc",
-    url = "https://github.com/grpc/grpc/archive/refs/tags/v1.50.0.tar.gz",
-    sha256 = "76900ab068da86378395a8e125b5cc43dfae671e09ff6462ddfef18676e2165a",
-    strip_prefix = "grpc-1.50.0",
+    patches = [
+        "//third_party/grpc:register_go_toolchain.patch",
+        "//third_party/grpc:grpc_noexcept.patch",
+    ],
+    sha256 = "0343e6dbde66e9a31c691f2f61e98d79f3584e03a11511fad3f10e3667832a45",
+    strip_prefix = "grpc-1.29.1",
+    url = "https://github.com/grpc/grpc/archive/refs/tags/v1.29.1.tar.gz",
+)
+
+http_archive(
+    name = "pybind11_protobuf",
+    sha256 = "ba2c54a8b4d1dd0a68c58159e37b1f863c0d9d1dc815558288195493bcc31682",
+    strip_prefix = "pybind11_protobuf-80f3440cd8fee124e077e2e47a8a17b78b451363",
+    url = "https://github.com/pybind/pybind11_protobuf/archive/80f3440cd8fee124e077e2e47a8a17b78b451363.tar.gz",
 )
 
 http_archive(
     name = "com_google_benchmark",
-    url = "https://github.com/google/benchmark/archive/refs/tags/v1.8.3.tar.gz",
     sha256 = "6bc180a57d23d4d9515519f92b0c83d61b05b5bab188961f36ac7b06b0d9e9ce",
     strip_prefix = "benchmark-1.8.3",
+    url = "https://github.com/google/benchmark/archive/refs/tags/v1.8.3.tar.gz",
 )
 
 http_archive(
-  name = "com_google_cc_differential_privacy",
-  url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
-  sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
-  strip_prefix = "differential-privacy-3.0.0/cc",
+    name = "com_google_cc_differential_privacy",
+    sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
+    strip_prefix = "differential-privacy-3.0.0/cc",
+    url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
 )
 
 http_archive(
-  name = "com_google_differential_privacy",
-  url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
-  sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
-  strip_prefix = "differential-privacy-3.0.0",
+    name = "com_google_differential_privacy",
+    sha256 = "6e6e1cd7a819695caae408f4fa938129ab7a86e83fe2410137c85e50131abbe0",
+    strip_prefix = "differential-privacy-3.0.0",
+    url = "https://github.com/google/differential-privacy/archive/refs/tags/v3.0.0.tar.gz",
 )
 
 # This commit is determined by
-# https://github.com/tensorflow/tensorflow/blob/v2.16.1/third_party/absl/workspace.bzl#L10.
+# https://github.com/tensorflow/tensorflow/blob/v2.17.0/third_party/absl/workspace.bzl#L10.
 http_archive(
     name = "com_google_absl",
-    url = "https://github.com/abseil/abseil-cpp/archive/fb3621f4f897824c0dbe0615fa94543df6192f30.tar.gz",
     sha256 = "0320586856674d16b0b7a4d4afb22151bdc798490bb7f295eddd8f6a62b46fea",
     strip_prefix = "abseil-cpp-fb3621f4f897824c0dbe0615fa94543df6192f30",
+    url = "https://github.com/abseil/abseil-cpp/archive/fb3621f4f897824c0dbe0615fa94543df6192f30.tar.gz",
 )
 
 http_archive(
     name = "com_google_googletest",
-    url = "https://github.com/google/googletest/archive/refs/tags/release-1.12.1.tar.gz",
     sha256 = "81964fe578e9bd7c94dfdb09c8e4d6e6759e19967e397dbea48d1c10e45d0df2",
     strip_prefix = "googletest-release-1.12.1",
+    url = "https://github.com/google/googletest/archive/refs/tags/release-1.12.1.tar.gz",
 )
 
 http_archive(
     name = "com_google_protobuf",
-    url = "https://github.com/protocolbuffers/protobuf/archive/refs/tags/v3.21.9.tar.gz",
     sha256 = "1add10f9bd92775b91f326da259f243881e904dd509367d5031d4c782ba82810",
     strip_prefix = "protobuf-3.21.9",
+    url = "https://github.com/protocolbuffers/protobuf/archive/refs/tags/v3.21.9.tar.gz",
 )
 
-# TODO: b/333391041 - Temporarily disable the direct dependency on
-# `eigen`, for now we pick this dependency up from the TensorFlow workspace.
+# TODO: b/333391041 - Temporarily disable the direct dependency on `eigen`, for
+# now we pick this dependency up from the TensorFlow workspace.
 # http_archive(
 #     name = "eigen",
 #     url = "https://gitlab.com/libeigen/eigen/archive/refs/tags/3.4.0.tar.gz",
@@ -91,55 +102,63 @@ http_archive(
 # https://github.com/google-parfait/tensorflow-federated/blob/main/requirements.txt.
 http_archive(
     name = "org_tensorflow",
-    url = "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.14.0.tar.gz",
     patches = [
         "//third_party/tensorflow:internal_visibility.patch",
-        "//third_party/tensorflow:python_toolchain.patch",
         "//third_party/tensorflow:tf2xla_visibility.patch",
     ],
-    sha256 = "ce357fd0728f0d1b0831d1653f475591662ec5bca736a94ff789e6b1944df19f",
-    strip_prefix = "tensorflow-2.14.0",
+    sha256 = "9cc4d5773b8ee910079baaecb4086d0c28939f024dd74b33fc5e64779b6533dc",
+    strip_prefix = "tensorflow-2.17.0",
+    url = "https://github.com/tensorflow/tensorflow/archive/refs/tags/v2.17.0.tar.gz",
 )
 
 # This commit is determined by
-# https://github.com/tensorflow/tensorflow/blob/v2.16.1/third_party/pybind11_abseil/workspace.bzl#L11.
+# https://github.com/tensorflow/tensorflow/blob/v2.17.0/third_party/pybind11_abseil/workspace.bzl#L11.
 http_archive(
     name = "pybind11_abseil",
-    url = "https://github.com/pybind/pybind11_abseil/archive/2c4932ed6f6204f1656e245838f4f5eae69d2e29.tar.gz",
     sha256 = "0223b647b8cc817336a51e787980ebc299c8d5e64c069829bf34b69d72337449",
     strip_prefix = "pybind11_abseil-2c4932ed6f6204f1656e245838f4f5eae69d2e29",
+    url = "https://github.com/pybind/pybind11_abseil/archive/2c4932ed6f6204f1656e245838f4f5eae69d2e29.tar.gz",
 )
 
 http_archive(
     name = "pybind11_bazel",
-    url = "https://github.com/pybind/pybind11_bazel/archive/refs/tags/v2.11.1.tar.gz",
     sha256 = "e8355ee56c2ff772334b4bfa22be17c709e5573f6d1d561c7176312156c27bd4",
     strip_prefix = "pybind11_bazel-2.11.1",
-)
-
-# This commit is determined by
-# https://github.com/tensorflow/tensorflow/blob/v2.16.1/tensorflow/workspace2.bzl#L788.
-http_archive(
-    name = "pybind11_protobuf",
-    url = "https://github.com/pybind/pybind11_protobuf/archive/80f3440cd8fee124e077e2e47a8a17b78b451363.tar.gz",
-    sha256 = "ba2c54a8b4d1dd0a68c58159e37b1f863c0d9d1dc815558288195493bcc31682",
-    strip_prefix = "pybind11_protobuf-80f3440cd8fee124e077e2e47a8a17b78b451363",
+    url = "https://github.com/pybind/pybind11_bazel/archive/refs/tags/v2.11.1.tar.gz",
 )
 
 http_archive(
     name = "rules_license",
-    url = "https://github.com/bazelbuild/rules_license/archive/refs/tags/0.0.8.tar.gz",
     sha256 = "8c1155797cb5f5697ea8c6eac6c154cf51aa020e368813d9d9b949558c84f2da",
     strip_prefix = "rules_license-0.0.8",
+    url = "https://github.com/bazelbuild/rules_license/archive/refs/tags/0.0.8.tar.gz",
 )
 
 http_archive(
     name = "rules_python",
-    url = "https://github.com/bazelbuild/rules_python/archive/refs/tags/0.31.0.tar.gz",
-    sha256 = "c68bdc4fbec25de5b5493b8819cfc877c4ea299c0dcb15c244c5a00208cde311",
-    strip_prefix = "rules_python-0.31.0",
+    sha256 = "778aaeab3e6cfd56d681c89f5c10d7ad6bf8d2f1a72de9de55b23081b2d31618",
+    strip_prefix = "rules_python-0.34.0",
+    url = "https://github.com/bazelbuild/rules_python/archive/refs/tags/0.34.0.tar.gz",
 )
 
+# TODO: b/347940678 - Temporarily disable the direct dependency on `tsl`, for
+# now we pick this dependency up from the TensorFlow workspace.
+# http_archive(
+#     name = "tsl",
+#     url = "https://github.com/google/tsl/archive/5fdb99d897eadddd136c41156fbda82c6244d60a.tar.gz",
+#     sha256 = "414f9929d6a631aa225093e24e657e1a28c5e84da8e280f1636db65fe24e4b83",
+#     strip_prefix = "tsl-5fdb99d897eadddd136c41156fbda82c6244d60a",
+# )
+
+# TODO: b/347937714 - Temporarily disable the direct dependency on `xla`, for
+# now we pick this dependency up from the TensorFlow workspace.
+# http_archive(
+#     name = "xla",
+#     url = "https://github.com/openxla/xla/archive/940e3a27542b7ce76666173e7b287aa2a9263916.tar.gz",
+#     sha256 = "bcdc778e5a456839869dea796117b723bdea488075bd9555fe118fd8d6fcf25e",
+#     strip_prefix = "xla-940e3a27542b7ce76666173e7b287aa2a9263916",
+# )
+
 #
 # Inlined transitive dependencies, grouped by direct dependency.
 #
@@ -147,20 +166,23 @@ http_archive(
 # Required by pybind11_abseil and pybind11_protobuf.
 http_archive(
     name = "pybind11",
-    url = "https://github.com/pybind/pybind11/archive/refs/tags/v2.9.2.tar.gz",
+    build_file = "@pybind11_bazel//:pybind11.BUILD",
     sha256 = "6bd528c4dbe2276635dc787b6b1f2e5316cf6b49ee3e150264e455a0d68d19c1",
     strip_prefix = "pybind11-2.9.2",
-    build_file = "@pybind11_bazel//:pybind11.BUILD",
+    url = "https://github.com/pybind/pybind11/archive/refs/tags/v2.9.2.tar.gz",
 )
 
+# TODO: b/347937669 - Temporarily disable the direct dependency on
+# `com_github_grpc_grpc`, for now we pick this dependency up from the
+# TensorFlow workspace.
 # Required by com_github_grpc_grpc. This commit is determined by
 # https://github.com/grpc/grpc/blob/v1.50.0/bazel/grpc_deps.bzl#L344.
-http_archive(
-    name = "upb",
-    url = "https://github.com/protocolbuffers/upb/archive/e4635f223e7d36dfbea3b722a4ca4807a7e882e2.tar.gz",
-    sha256 = "017a7e8e4e842d01dba5dc8aa316323eee080cd1b75986a7d1f94d87220e6502",
-    strip_prefix = "upb-e4635f223e7d36dfbea3b722a4ca4807a7e882e2",
-)
+# http_archive(
+#     name = "upb",
+#     url = "https://github.com/protocolbuffers/upb/archive/e4635f223e7d36dfbea3b722a4ca4807a7e882e2.tar.gz",
+#     sha256 = "017a7e8e4e842d01dba5dc8aa316323eee080cd1b75986a7d1f94d87220e6502",
+#     strip_prefix = "upb-e4635f223e7d36dfbea3b722a4ca4807a7e882e2",
+# )
 
 #
 # Transitive dependencies, grouped by direct dependency.
@@ -170,6 +192,27 @@ load("@rules_python//python:repositories.bzl", "py_repositories")
 
 py_repositories()
 
+# Initialize hermetic Python
+load("@org_tensorflow//third_party/py:python_init_rules.bzl", "python_init_rules")
+
+python_init_rules()
+
+load("@org_tensorflow//third_party/py:python_init_repositories.bzl", "python_init_repositories")
+
+python_init_repositories(requirements = {"3.11": "//:requirements_lock_3_11.txt"})
+
+load("@org_tensorflow//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")
+
+python_init_toolchains()
+
+load("@org_tensorflow//third_party/py:python_init_pip.bzl", "python_init_pip")
+
+python_init_pip()
+
+load("@pypi//:requirements.bzl", "install_deps")
+
+install_deps()
+
 load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3")
 
 tf_workspace3()
diff --git a/examples/custom_data_backend/BUILD b/examples/custom_data_backend/BUILD
index 5bd10a35c9..47d21a78a5 100644
--- a/examples/custom_data_backend/BUILD
+++ b/examples/custom_data_backend/BUILD
@@ -1,6 +1,8 @@
+load("//tools:build_defs.bzl", "tff_cc_library")
+
 licenses(["notice"])
 
-cc_library(
+tff_cc_library(
     name = "data_backend_example_cc",
     srcs = ["data_backend_example.cc"],
     hdrs = ["data_backend_example.h"],
diff --git a/examples/datasets/BUILD b/examples/datasets/BUILD
index b5269d917d..6f5aa86578 100644
--- a/examples/datasets/BUILD
+++ b/examples/datasets/BUILD
@@ -10,5 +10,8 @@ licenses(["notice"])
 py_binary(
     name = "load_flair",
     srcs = ["load_flair.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+    ],
 )
diff --git a/examples/learning/federated_program/vizier/BUILD b/examples/learning/federated_program/vizier/BUILD
index 517888071a..384451b873 100644
--- a/examples/learning/federated_program/vizier/BUILD
+++ b/examples/learning/federated_program/vizier/BUILD
@@ -3,13 +3,19 @@ load("@rules_python//python:defs.bzl", "py_binary", "py_library", "py_test")
 py_library(
     name = "data_sources",
     srcs = ["data_sources.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "learning_process",
     srcs = ["learning_process.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_binary(
@@ -23,12 +29,15 @@ py_binary(
         ":learning_process",
         ":vizier_service",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_google_vizier//:pkg",
     ],
 )
 
 py_library(
     name = "vizier_service",
     srcs = ["vizier_service.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
@@ -37,5 +46,8 @@ py_test(
     data = [
         ":study_spec.textproto",
     ],
-    deps = [":vizier_service"],
+    deps = [
+        ":vizier_service",
+        "@pypi_absl_py//:pkg",
+    ],
 )
diff --git a/examples/personalization/BUILD b/examples/personalization/BUILD
index 79b03a8d71..1dd2f4172c 100644
--- a/examples/personalization/BUILD
+++ b/examples/personalization/BUILD
@@ -10,7 +10,10 @@ licenses(["notice"])
 py_library(
     name = "p13n_utils",
     srcs = ["p13n_utils.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -19,6 +22,8 @@ py_test(
     deps = [
         ":p13n_utils",
         "//tensorflow_federated",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -28,5 +33,8 @@ py_binary(
     deps = [
         ":p13n_utils",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/examples/program/BUILD b/examples/program/BUILD
index 7bd07db1c7..327e74b00a 100644
--- a/examples/program/BUILD
+++ b/examples/program/BUILD
@@ -10,7 +10,11 @@ licenses(["notice"])
 py_library(
     name = "computations",
     srcs = ["computations.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -26,6 +30,8 @@ py_binary(
         ":computations",
         ":program_logic",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -36,5 +42,8 @@ py_test(
         ":computations",
         ":program_logic",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/examples/simple_fedavg/BUILD b/examples/simple_fedavg/BUILD
index bbccce128c..e4f818b454 100644
--- a/examples/simple_fedavg/BUILD
+++ b/examples/simple_fedavg/BUILD
@@ -13,13 +13,21 @@ py_binary(
     deps = [
         ":simple_fedavg_tff",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "simple_fedavg_tf",
     srcs = ["simple_fedavg_tf.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -28,6 +36,7 @@ py_library(
     deps = [
         ":simple_fedavg_tf",
         "//tensorflow_federated",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -41,5 +50,9 @@ py_test(
         ":simple_fedavg_tf",
         ":simple_fedavg_tff",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/examples/stateful_clients/BUILD b/examples/stateful_clients/BUILD
index 05cc2bb269..fabeae1ff3 100644
--- a/examples/stateful_clients/BUILD
+++ b/examples/stateful_clients/BUILD
@@ -14,13 +14,20 @@ py_binary(
         ":stateful_fedavg_tf",
         ":stateful_fedavg_tff",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "stateful_fedavg_tf",
     srcs = ["stateful_fedavg_tf.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_attrs//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -29,6 +36,8 @@ py_library(
     deps = [
         ":stateful_fedavg_tf",
         "//tensorflow_federated",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -40,5 +49,8 @@ py_test(
         ":stateful_fedavg_tf",
         ":stateful_fedavg_tff",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/pyproject.toml b/pyproject.toml
index bcd14f72b9..c0bb397337 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -46,7 +46,7 @@ dependencies = [
   'scipy~=1.9.3',
   'tensorflow-model-optimization==0.7.5',
   'tensorflow-privacy==0.9.0',
-  'tensorflow>=2.14.0,==2.14.*',
+  'tensorflow~=2.16.1',
   'tqdm~=4.64',
   'typing-extensions>=4.5.0,==4.5.*',
   # TODO: b/315515548 - Required because current dependencies are pulling in
diff --git a/requirements.txt b/requirements.in
similarity index 98%
rename from requirements.txt
rename to requirements.in
index cc247f214d..579b8e7177 100644
--- a/requirements.txt
+++ b/requirements.in
@@ -31,7 +31,7 @@ cachetools~=5.3
 dm-tree==0.1.8
 dp-accounting==0.4.3
 google-vizier==0.1.11
-grpcio~=1.46
+grpcio~=1.64
 jaxlib==0.4.14
 jax==0.4.14
 numpy~=1.25
@@ -44,7 +44,7 @@ typing-extensions>=4.5.0,==4.5.*
 
 # The version of this dependency should match the version in
 # https://github.com/google-parfait/tensorflow-federated/blob/main/WORKSPACE.
-tensorflow>=2.14.0,==2.14.*
+tensorflow~=2.17.0
 
 # TODO: b/315515548 - Required because current dependencies are pulling in later
 # versions which are causing opaque failures.
diff --git a/requirements_lock_3_11.txt b/requirements_lock_3_11.txt
new file mode 100644
index 0000000000..a1ed9538ac
--- /dev/null
+++ b/requirements_lock_3_11.txt
@@ -0,0 +1,1007 @@
+#
+# This file is autogenerated by pip-compile with Python 3.11
+# by the following command:
+#
+#    bazel run //:requirements.update
+#
+absl-py==1.4.0 \
+    --hash=sha256:0d3fe606adfa4f7db64792dd4c7aee4ee0c38ab75dfd353b7a83ed3e957fcb47 \
+    --hash=sha256:d2c244d01048ba476e7c080bd2c6df5e141d211de80223460d5b3b8a2a58433d
+    # via
+    #   -r requirements.in
+    #   dp-accounting
+    #   google-vizier
+    #   keras
+    #   tensorboard
+    #   tensorflow
+    #   tensorflow-model-optimization
+    #   tensorflow-privacy
+    #   tensorflow-probability
+astunparse==1.6.3 \
+    --hash=sha256:5ad93a8456f0d084c3456d059fd9a92cce667963232cbf763eac3bc5b7940872 \
+    --hash=sha256:c2652417f2c8b5bb325c885ae329bdf3f86424075c4fd1a128674bc6fba4b8e8
+    # via tensorflow
+attrs==23.1.0 \
+    --hash=sha256:1f28b4522cdc2fb4256ac1a020c78acf9cba2c6b461ccd2c126f3aa8e8335d04 \
+    --hash=sha256:6279836d581513a26f1bf235f9acd333bc9115683f14f7e8fae46c98fc50e015
+    # via
+    #   -r requirements.in
+    #   dp-accounting
+    #   google-vizier
+cachetools==5.4.0 \
+    --hash=sha256:3ae3b49a3d5e28a77a0be2b37dbcb89005058959cb2323858c2657c4a8cab474 \
+    --hash=sha256:b8adc2e7c07f105ced7bc56dbb6dfbe7c4a00acce20e2227b3f355be89bc6827
+    # via -r requirements.in
+certifi==2024.7.4 \
+    --hash=sha256:5a1e7645bc0ec61a09e26c36f6106dd4cf40c6db3a1fb6352b0244e7fb057c7b \
+    --hash=sha256:c198e21b1289c2ab85ee4e67bb4b4ef3ead0892059901a8d5b622f24a1101e90
+    # via requests
+charset-normalizer==3.3.2 \
+    --hash=sha256:06435b539f889b1f6f4ac1758871aae42dc3a8c0e24ac9e60c2384973ad73027 \
+    --hash=sha256:06a81e93cd441c56a9b65d8e1d043daeb97a3d0856d177d5c90ba85acb3db087 \
+    --hash=sha256:0a55554a2fa0d408816b3b5cedf0045f4b8e1a6065aec45849de2d6f3f8e9786 \
+    --hash=sha256:0b2b64d2bb6d3fb9112bafa732def486049e63de9618b5843bcdd081d8144cd8 \
+    --hash=sha256:10955842570876604d404661fbccbc9c7e684caf432c09c715ec38fbae45ae09 \
+    --hash=sha256:122c7fa62b130ed55f8f285bfd56d5f4b4a5b503609d181f9ad85e55c89f4185 \
+    --hash=sha256:1ceae2f17a9c33cb48e3263960dc5fc8005351ee19db217e9b1bb15d28c02574 \
+    --hash=sha256:1d3193f4a680c64b4b6a9115943538edb896edc190f0b222e73761716519268e \
+    --hash=sha256:1f79682fbe303db92bc2b1136016a38a42e835d932bab5b3b1bfcfbf0640e519 \
+    --hash=sha256:2127566c664442652f024c837091890cb1942c30937add288223dc895793f898 \
+    --hash=sha256:22afcb9f253dac0696b5a4be4a1c0f8762f8239e21b99680099abd9b2b1b2269 \
+    --hash=sha256:25baf083bf6f6b341f4121c2f3c548875ee6f5339300e08be3f2b2ba1721cdd3 \
+    --hash=sha256:2e81c7b9c8979ce92ed306c249d46894776a909505d8f5a4ba55b14206e3222f \
+    --hash=sha256:3287761bc4ee9e33561a7e058c72ac0938c4f57fe49a09eae428fd88aafe7bb6 \
+    --hash=sha256:34d1c8da1e78d2e001f363791c98a272bb734000fcef47a491c1e3b0505657a8 \
+    --hash=sha256:37e55c8e51c236f95b033f6fb391d7d7970ba5fe7ff453dad675e88cf303377a \
+    --hash=sha256:3d47fa203a7bd9c5b6cee4736ee84ca03b8ef23193c0d1ca99b5089f72645c73 \
+    --hash=sha256:3e4d1f6587322d2788836a99c69062fbb091331ec940e02d12d179c1d53e25fc \
+    --hash=sha256:42cb296636fcc8b0644486d15c12376cb9fa75443e00fb25de0b8602e64c1714 \
+    --hash=sha256:45485e01ff4d3630ec0d9617310448a8702f70e9c01906b0d0118bdf9d124cf2 \
+    --hash=sha256:4a78b2b446bd7c934f5dcedc588903fb2f5eec172f3d29e52a9096a43722adfc \
+    --hash=sha256:4ab2fe47fae9e0f9dee8c04187ce5d09f48eabe611be8259444906793ab7cbce \
+    --hash=sha256:4d0d1650369165a14e14e1e47b372cfcb31d6ab44e6e33cb2d4e57265290044d \
+    --hash=sha256:549a3a73da901d5bc3ce8d24e0600d1fa85524c10287f6004fbab87672bf3e1e \
+    --hash=sha256:55086ee1064215781fff39a1af09518bc9255b50d6333f2e4c74ca09fac6a8f6 \
+    --hash=sha256:572c3763a264ba47b3cf708a44ce965d98555f618ca42c926a9c1616d8f34269 \
+    --hash=sha256:573f6eac48f4769d667c4442081b1794f52919e7edada77495aaed9236d13a96 \
+    --hash=sha256:5b4c145409bef602a690e7cfad0a15a55c13320ff7a3ad7ca59c13bb8ba4d45d \
+    --hash=sha256:6463effa3186ea09411d50efc7d85360b38d5f09b870c48e4600f63af490e56a \
+    --hash=sha256:65f6f63034100ead094b8744b3b97965785388f308a64cf8d7c34f2f2e5be0c4 \
+    --hash=sha256:663946639d296df6a2bb2aa51b60a2454ca1cb29835324c640dafb5ff2131a77 \
+    --hash=sha256:6897af51655e3691ff853668779c7bad41579facacf5fd7253b0133308cf000d \
+    --hash=sha256:68d1f8a9e9e37c1223b656399be5d6b448dea850bed7d0f87a8311f1ff3dabb0 \
+    --hash=sha256:6ac7ffc7ad6d040517be39eb591cac5ff87416c2537df6ba3cba3bae290c0fed \
+    --hash=sha256:6b3251890fff30ee142c44144871185dbe13b11bab478a88887a639655be1068 \
+    --hash=sha256:6c4caeef8fa63d06bd437cd4bdcf3ffefe6738fb1b25951440d80dc7df8c03ac \
+    --hash=sha256:6ef1d82a3af9d3eecdba2321dc1b3c238245d890843e040e41e470ffa64c3e25 \
+    --hash=sha256:753f10e867343b4511128c6ed8c82f7bec3bd026875576dfd88483c5c73b2fd8 \
+    --hash=sha256:7cd13a2e3ddeed6913a65e66e94b51d80a041145a026c27e6bb76c31a853c6ab \
+    --hash=sha256:7ed9e526742851e8d5cc9e6cf41427dfc6068d4f5a3bb03659444b4cabf6bc26 \
+    --hash=sha256:7f04c839ed0b6b98b1a7501a002144b76c18fb1c1850c8b98d458ac269e26ed2 \
+    --hash=sha256:802fe99cca7457642125a8a88a084cef28ff0cf9407060f7b93dca5aa25480db \
+    --hash=sha256:80402cd6ee291dcb72644d6eac93785fe2c8b9cb30893c1af5b8fdd753b9d40f \
+    --hash=sha256:8465322196c8b4d7ab6d1e049e4c5cb460d0394da4a27d23cc242fbf0034b6b5 \
+    --hash=sha256:86216b5cee4b06df986d214f664305142d9c76df9b6512be2738aa72a2048f99 \
+    --hash=sha256:87d1351268731db79e0f8e745d92493ee2841c974128ef629dc518b937d9194c \
+    --hash=sha256:8bdb58ff7ba23002a4c5808d608e4e6c687175724f54a5dade5fa8c67b604e4d \
+    --hash=sha256:8c622a5fe39a48f78944a87d4fb8a53ee07344641b0562c540d840748571b811 \
+    --hash=sha256:8d756e44e94489e49571086ef83b2bb8ce311e730092d2c34ca8f7d925cb20aa \
+    --hash=sha256:8f4a014bc36d3c57402e2977dada34f9c12300af536839dc38c0beab8878f38a \
+    --hash=sha256:9063e24fdb1e498ab71cb7419e24622516c4a04476b17a2dab57e8baa30d6e03 \
+    --hash=sha256:90d558489962fd4918143277a773316e56c72da56ec7aa3dc3dbbe20fdfed15b \
+    --hash=sha256:923c0c831b7cfcb071580d3f46c4baf50f174be571576556269530f4bbd79d04 \
+    --hash=sha256:95f2a5796329323b8f0512e09dbb7a1860c46a39da62ecb2324f116fa8fdc85c \
+    --hash=sha256:96b02a3dc4381e5494fad39be677abcb5e6634bf7b4fa83a6dd3112607547001 \
+    --hash=sha256:9f96df6923e21816da7e0ad3fd47dd8f94b2a5ce594e00677c0013018b813458 \
+    --hash=sha256:a10af20b82360ab00827f916a6058451b723b4e65030c5a18577c8b2de5b3389 \
+    --hash=sha256:a50aebfa173e157099939b17f18600f72f84eed3049e743b68ad15bd69b6bf99 \
+    --hash=sha256:a981a536974bbc7a512cf44ed14938cf01030a99e9b3a06dd59578882f06f985 \
+    --hash=sha256:a9a8e9031d613fd2009c182b69c7b2c1ef8239a0efb1df3f7c8da66d5dd3d537 \
+    --hash=sha256:ae5f4161f18c61806f411a13b0310bea87f987c7d2ecdbdaad0e94eb2e404238 \
+    --hash=sha256:aed38f6e4fb3f5d6bf81bfa990a07806be9d83cf7bacef998ab1a9bd660a581f \
+    --hash=sha256:b01b88d45a6fcb69667cd6d2f7a9aeb4bf53760d7fc536bf679ec94fe9f3ff3d \
+    --hash=sha256:b261ccdec7821281dade748d088bb6e9b69e6d15b30652b74cbbac25e280b796 \
+    --hash=sha256:b2b0a0c0517616b6869869f8c581d4eb2dd83a4d79e0ebcb7d373ef9956aeb0a \
+    --hash=sha256:b4a23f61ce87adf89be746c8a8974fe1c823c891d8f86eb218bb957c924bb143 \
+    --hash=sha256:bd8f7df7d12c2db9fab40bdd87a7c09b1530128315d047a086fa3ae3435cb3a8 \
+    --hash=sha256:beb58fe5cdb101e3a055192ac291b7a21e3b7ef4f67fa1d74e331a7f2124341c \
+    --hash=sha256:c002b4ffc0be611f0d9da932eb0f704fe2602a9a949d1f738e4c34c75b0863d5 \
+    --hash=sha256:c083af607d2515612056a31f0a8d9e0fcb5876b7bfc0abad3ecd275bc4ebc2d5 \
+    --hash=sha256:c180f51afb394e165eafe4ac2936a14bee3eb10debc9d9e4db8958fe36afe711 \
+    --hash=sha256:c235ebd9baae02f1b77bcea61bce332cb4331dc3617d254df3323aa01ab47bd4 \
+    --hash=sha256:cd70574b12bb8a4d2aaa0094515df2463cb429d8536cfb6c7ce983246983e5a6 \
+    --hash=sha256:d0eccceffcb53201b5bfebb52600a5fb483a20b61da9dbc885f8b103cbe7598c \
+    --hash=sha256:d965bba47ddeec8cd560687584e88cf699fd28f192ceb452d1d7ee807c5597b7 \
+    --hash=sha256:db364eca23f876da6f9e16c9da0df51aa4f104a972735574842618b8c6d999d4 \
+    --hash=sha256:ddbb2551d7e0102e7252db79ba445cdab71b26640817ab1e3e3648dad515003b \
+    --hash=sha256:deb6be0ac38ece9ba87dea880e438f25ca3eddfac8b002a2ec3d9183a454e8ae \
+    --hash=sha256:e06ed3eb3218bc64786f7db41917d4e686cc4856944f53d5bdf83a6884432e12 \
+    --hash=sha256:e27ad930a842b4c5eb8ac0016b0a54f5aebbe679340c26101df33424142c143c \
+    --hash=sha256:e537484df0d8f426ce2afb2d0f8e1c3d0b114b83f8850e5f2fbea0e797bd82ae \
+    --hash=sha256:eb00ed941194665c332bf8e078baf037d6c35d7c4f3102ea2d4f16ca94a26dc8 \
+    --hash=sha256:eb6904c354526e758fda7167b33005998fb68c46fbc10e013ca97f21ca5c8887 \
+    --hash=sha256:eb8821e09e916165e160797a6c17edda0679379a4be5c716c260e836e122f54b \
+    --hash=sha256:efcb3f6676480691518c177e3b465bcddf57cea040302f9f4e6e191af91174d4 \
+    --hash=sha256:f27273b60488abe721a075bcca6d7f3964f9f6f067c8c4c605743023d7d3944f \
+    --hash=sha256:f30c3cb33b24454a82faecaf01b19c18562b1e89558fb6c56de4d9118a032fd5 \
+    --hash=sha256:fb69256e180cb6c8a894fee62b3afebae785babc1ee98b81cdf68bbca1987f33 \
+    --hash=sha256:fd1abc0d89e30cc4e02e4064dc67fcc51bd941eb395c502aac3ec19fab46b519 \
+    --hash=sha256:ff8fa367d09b717b2a17a052544193ad76cd49979c805768879cb63d9ca50561
+    # via requests
+cloudpickle==3.0.0 \
+    --hash=sha256:246ee7d0c295602a036e86369c77fecda4ab17b506496730f2f576d9016fd9c7 \
+    --hash=sha256:996d9a482c6fb4f33c1a35335cf8afd065d2a56e973270364840712d9131a882
+    # via tensorflow-probability
+decorator==5.1.1 \
+    --hash=sha256:637996211036b6385ef91435e4fae22989472f9d571faba8927ba8253acbc330 \
+    --hash=sha256:b8c3f85900b9dc423225913c5aace94729fe1fa9763b38939a95226f02d37186
+    # via tensorflow-probability
+dm-tree==0.1.8 \
+    --hash=sha256:054b461f8176f4bce7a21f7b1870f873a1ced3bdbe1282c816c550bb43c71fa6 \
+    --hash=sha256:09964470f76a5201aff2e8f9b26842976de7889300676f927930f6285e256760 \
+    --hash=sha256:0d3172394079a86c3a759179c65f64c48d1a42b89495fcf38976d11cc3bb952c \
+    --hash=sha256:0e9620ccf06393eb6b613b5e366469304622d4ea96ae6540b28a33840e6c89cf \
+    --hash=sha256:0fcaabbb14e7980377439e7140bd05552739ca5e515ecb3119f234acee4b9430 \
+    --hash=sha256:1607ce49aa42f010d1e5e616d92ce899d66835d4d8bea49679582435285515de \
+    --hash=sha256:181c35521d480d0365f39300542cb6cd7fd2b77351bb43d7acfda15aef63b317 \
+    --hash=sha256:1d7c26e431fc93cc7e0cba867eb000db6a05f6f2b25af11ac4e9dada88fc5bca \
+    --hash=sha256:1fe962015b2fe1282892b28ebe962faed53c7f98d942da9a4625cbf27baef913 \
+    --hash=sha256:250b692fb75f45f02e2f58fbef9ab338904ef334b90557565621fa251df267cf \
+    --hash=sha256:2869228d9c619074de501a3c10dc7f07c75422f8fab36ecdcb859b6f1b1ec3ef \
+    --hash=sha256:28c52cbf4f8b3dbd0beaedf44f69fa85eec5e9dede612e08035e06ada6ec9426 \
+    --hash=sha256:2f7915660f59c09068e428613c480150180df1060561fd0d1470684ae7007bd1 \
+    --hash=sha256:343a4a4ebaa127451ff971254a4be4084eb4bdc0b2513c32b46f6f728fd03f9e \
+    --hash=sha256:35cc164a79336bfcfafb47e5f297898359123bbd3330c1967f0c4994f9cf9f60 \
+    --hash=sha256:378cc8ad93c5fe3590f405a309980721f021c790ca1bdf9b15bb1d59daec57f5 \
+    --hash=sha256:39070ba268c0491af9fe7a58644d99e8b4f2cde6e5884ba3380bddc84ed43d5f \
+    --hash=sha256:435227cf3c5dc63f4de054cf3d00183790bd9ead4c3623138c74dde7f67f521b \
+    --hash=sha256:5483dca4d7eb1a0d65fe86d3b6a53ae717face83c1f17e0887b1a4a64ae5c410 \
+    --hash=sha256:694c3654cfd2a81552c08ec66bb5c4a3d48fa292b9a181880fb081c36c5b9134 \
+    --hash=sha256:75c5d528bb992981c20793b6b453e91560784215dffb8a5440ba999753c14ceb \
+    --hash=sha256:803bfc53b4659f447ac694dbd04235f94a73ef7c1fd1e0df7c84ac41e0bc963b \
+    --hash=sha256:81fce77f22a302d7a5968aebdf4efafef4def7ce96528719a354e6990dcd49c7 \
+    --hash=sha256:83b7764de0d855338abefc6e3ee9fe40d301668310aa3baea3f778ff051f4393 \
+    --hash=sha256:8c60a7eadab64c2278861f56bca320b2720f163dca9d7558103c3b77f2416571 \
+    --hash=sha256:8ed3564abed97c806db122c2d3e1a2b64c74a63debe9903aad795167cc301368 \
+    --hash=sha256:94d3f0826311f45ee19b75f5b48c99466e4218a0489e81c0f0167bda50cacf22 \
+    --hash=sha256:96a548a406a6fb15fe58f6a30a57ff2f2aafbf25f05afab00c8f5e5977b6c715 \
+    --hash=sha256:a5d819c38c03f0bb5b3b3703c60e4b170355a0fc6b5819325bf3d4ceb3ae7e80 \
+    --hash=sha256:ad16ceba90a56ec47cf45b21856d14962ac314787975ef786efb5e6e9ca75ec7 \
+    --hash=sha256:af4b3d372f2477dcd89a6e717e4a575ca35ccc20cc4454a8a4b6f8838a00672d \
+    --hash=sha256:b095ba4f8ca1ba19350fd53cf1f8f3eb0bd406aa28af64a6dfc86707b32a810a \
+    --hash=sha256:b9bd9b9ccb59409d33d51d84b7668010c04c2af7d4a371632874c1ca356cff3d \
+    --hash=sha256:b9f89a454e98806b44fe9d40ec9eee61f848388f7e79ac2371a55679bd5a3ac6 \
+    --hash=sha256:bb2d109f42190225112da899b9f3d46d0d5f26aef501c61e43529fe9322530b5 \
+    --hash=sha256:c0a94aba18a35457a1b5cd716fd7b46c5dafdc4cf7869b4bae665b91c4682a8e \
+    --hash=sha256:c5c8c12e3fda754ef6af94161bacdaeda816d941995fac415d6855c6c386af68 \
+    --hash=sha256:d1612fcaecd79023dbc6a6ae48d51a80beb5c385d6f3f6d71688e57bc8d07de8 \
+    --hash=sha256:d16e1f2a073604cfcc09f7131ae8d534674f43c3aef4c25742eae295bc60d04f \
+    --hash=sha256:d20f2faa3672b52e5013f4077117bfb99c4cfc0b445d3bde1584c34032b57436 \
+    --hash=sha256:d40fa4106ca6edc66760246a08f500ec0c85ef55c762fb4a363f6ee739ba02ee \
+    --hash=sha256:de287fabc464b8734be251e46e06aa9aa1001f34198da2b6ce07bd197172b9cb \
+    --hash=sha256:e4d714371bb08839e4e5e29024fc95832d9affe129825ef38836b143028bd144 \
+    --hash=sha256:ea9e59e0451e7d29aece402d9f908f2e2a80922bcde2ebfd5dcb07750fcbfee8 \
+    --hash=sha256:f7ac31b9aecccb2c6e1ab29706f6ded3eba0c2c69c770322c9c685929c3d6afb \
+    --hash=sha256:fa42a605d099ee7d41ba2b5fb75e21423951fd26e5d50583a00471238fb3021d
+    # via
+    #   -r requirements.in
+    #   dp-accounting
+    #   tensorflow-model-optimization
+    #   tensorflow-privacy
+    #   tensorflow-probability
+dp-accounting==0.4.3 \
+    --hash=sha256:bf7ffc4b8048a0bc44c3601a263a23dc6c4c37bed348f1dfe378eb808ceb3419 \
+    --hash=sha256:ef231a18729f8466fe62d6bfae2a243c5ed23e57df8bd84635e041e33f0d74b9
+    # via
+    #   -r requirements.in
+    #   tensorflow-privacy
+flatbuffers==24.3.25 \
+    --hash=sha256:8dbdec58f935f3765e4f7f3cf635ac3a77f83568138d6a2311f524ec96364812 \
+    --hash=sha256:de2ec5b203f21441716617f38443e0a8ebf3d25bf0d9c0bb0ce68fa00ad546a4
+    # via tensorflow
+gast==0.6.0 \
+    --hash=sha256:52b182313f7330389f72b069ba00f174cfe2a06411099547288839c6cbafbd54 \
+    --hash=sha256:88fc5300d32c7ac6ca7b515310862f71e6fdf2c029bbec7c66c0f5dd47b6b1fb
+    # via
+    #   tensorflow
+    #   tensorflow-probability
+google-pasta==0.2.0 \
+    --hash=sha256:4612951da876b1a10fe3960d7226f0c7682cf901e16ac06e473b267a5afa8954 \
+    --hash=sha256:b32482794a366b5366a32c92a9a9201b107821889935a02b3e51f6b432ea84ed \
+    --hash=sha256:c9f2c8dfc8f96d0d5808299920721be30c9eec37f2389f28904f454565c8a16e
+    # via tensorflow
+google-vizier==0.1.11 \
+    --hash=sha256:05e163e842679cbad066cab548af5f211b29e1b3b355121e85dc02e96fe616ee \
+    --hash=sha256:8c18eb44a37d596a0620997334deea25da7816e4e01290811a8f087b83fea93a
+    # via -r requirements.in
+googleapis-common-protos==1.61.0 \
+    --hash=sha256:22f1915393bb3245343f6efe87f6fe868532efc12aa26b391b15132e1279f1c0 \
+    --hash=sha256:8a64866a97f6304a7179873a465d6eee97b7a24ec6cfd78e0f575e96b821240b
+    # via
+    #   -r requirements.in
+    #   google-vizier
+greenlet==3.0.3 \
+    --hash=sha256:01bc7ea167cf943b4c802068e178bbf70ae2e8c080467070d01bfa02f337ee67 \
+    --hash=sha256:0448abc479fab28b00cb472d278828b3ccca164531daab4e970a0458786055d6 \
+    --hash=sha256:086152f8fbc5955df88382e8a75984e2bb1c892ad2e3c80a2508954e52295257 \
+    --hash=sha256:098d86f528c855ead3479afe84b49242e174ed262456c342d70fc7f972bc13c4 \
+    --hash=sha256:149e94a2dd82d19838fe4b2259f1b6b9957d5ba1b25640d2380bea9c5df37676 \
+    --hash=sha256:1551a8195c0d4a68fac7a4325efac0d541b48def35feb49d803674ac32582f61 \
+    --hash=sha256:15d79dd26056573940fcb8c7413d84118086f2ec1a8acdfa854631084393efcc \
+    --hash=sha256:1996cb9306c8595335bb157d133daf5cf9f693ef413e7673cb07e3e5871379ca \
+    --hash=sha256:1a7191e42732df52cb5f39d3527217e7ab73cae2cb3694d241e18f53d84ea9a7 \
+    --hash=sha256:1ea188d4f49089fc6fb283845ab18a2518d279c7cd9da1065d7a84e991748728 \
+    --hash=sha256:1f672519db1796ca0d8753f9e78ec02355e862d0998193038c7073045899f305 \
+    --hash=sha256:2516a9957eed41dd8f1ec0c604f1cdc86758b587d964668b5b196a9db5bfcde6 \
+    --hash=sha256:2797aa5aedac23af156bbb5a6aa2cd3427ada2972c828244eb7d1b9255846379 \
+    --hash=sha256:2dd6e660effd852586b6a8478a1d244b8dc90ab5b1321751d2ea15deb49ed414 \
+    --hash=sha256:3ddc0f794e6ad661e321caa8d2f0a55ce01213c74722587256fb6566049a8b04 \
+    --hash=sha256:3ed7fb269f15dc662787f4119ec300ad0702fa1b19d2135a37c2c4de6fadfd4a \
+    --hash=sha256:419b386f84949bf0e7c73e6032e3457b82a787c1ab4a0e43732898a761cc9dbf \
+    --hash=sha256:43374442353259554ce33599da8b692d5aa96f8976d567d4badf263371fbe491 \
+    --hash=sha256:52f59dd9c96ad2fc0d5724107444f76eb20aaccb675bf825df6435acb7703559 \
+    --hash=sha256:57e8974f23e47dac22b83436bdcf23080ade568ce77df33159e019d161ce1d1e \
+    --hash=sha256:5b51e85cb5ceda94e79d019ed36b35386e8c37d22f07d6a751cb659b180d5274 \
+    --hash=sha256:649dde7de1a5eceb258f9cb00bdf50e978c9db1b996964cd80703614c86495eb \
+    --hash=sha256:64d7675ad83578e3fc149b617a444fab8efdafc9385471f868eb5ff83e446b8b \
+    --hash=sha256:68834da854554926fbedd38c76e60c4a2e3198c6fbed520b106a8986445caaf9 \
+    --hash=sha256:6b66c9c1e7ccabad3a7d037b2bcb740122a7b17a53734b7d72a344ce39882a1b \
+    --hash=sha256:70fb482fdf2c707765ab5f0b6655e9cfcf3780d8d87355a063547b41177599be \
+    --hash=sha256:7170375bcc99f1a2fbd9c306f5be8764eaf3ac6b5cb968862cad4c7057756506 \
+    --hash=sha256:73a411ef564e0e097dbe7e866bb2dda0f027e072b04da387282b02c308807405 \
+    --hash=sha256:77457465d89b8263bca14759d7c1684df840b6811b2499838cc5b040a8b5b113 \
+    --hash=sha256:7f362975f2d179f9e26928c5b517524e89dd48530a0202570d55ad6ca5d8a56f \
+    --hash=sha256:81bb9c6d52e8321f09c3d165b2a78c680506d9af285bfccbad9fb7ad5a5da3e5 \
+    --hash=sha256:881b7db1ebff4ba09aaaeae6aa491daeb226c8150fc20e836ad00041bcb11230 \
+    --hash=sha256:894393ce10ceac937e56ec00bb71c4c2f8209ad516e96033e4b3b1de270e200d \
+    --hash=sha256:99bf650dc5d69546e076f413a87481ee1d2d09aaaaaca058c9251b6d8c14783f \
+    --hash=sha256:9da2bd29ed9e4f15955dd1595ad7bc9320308a3b766ef7f837e23ad4b4aac31a \
+    --hash=sha256:afaff6cf5200befd5cec055b07d1c0a5a06c040fe5ad148abcd11ba6ab9b114e \
+    --hash=sha256:b1b5667cced97081bf57b8fa1d6bfca67814b0afd38208d52538316e9422fc61 \
+    --hash=sha256:b37eef18ea55f2ffd8f00ff8fe7c8d3818abd3e25fb73fae2ca3b672e333a7a6 \
+    --hash=sha256:b542be2440edc2d48547b5923c408cbe0fc94afb9f18741faa6ae970dbcb9b6d \
+    --hash=sha256:b7dcbe92cc99f08c8dd11f930de4d99ef756c3591a5377d1d9cd7dd5e896da71 \
+    --hash=sha256:b7f009caad047246ed379e1c4dbcb8b020f0a390667ea74d2387be2998f58a22 \
+    --hash=sha256:bba5387a6975598857d86de9eac14210a49d554a77eb8261cc68b7d082f78ce2 \
+    --hash=sha256:c5e1536de2aad7bf62e27baf79225d0d64360d4168cf2e6becb91baf1ed074f3 \
+    --hash=sha256:c5ee858cfe08f34712f548c3c363e807e7186f03ad7a5039ebadb29e8c6be067 \
+    --hash=sha256:c9db1c18f0eaad2f804728c67d6c610778456e3e1cc4ab4bbd5eeb8e6053c6fc \
+    --hash=sha256:d353cadd6083fdb056bb46ed07e4340b0869c305c8ca54ef9da3421acbdf6881 \
+    --hash=sha256:d46677c85c5ba00a9cb6f7a00b2bfa6f812192d2c9f7d9c4f6a55b60216712f3 \
+    --hash=sha256:d4d1ac74f5c0c0524e4a24335350edad7e5f03b9532da7ea4d3c54d527784f2e \
+    --hash=sha256:d73a9fe764d77f87f8ec26a0c85144d6a951a6c438dfe50487df5595c6373eac \
+    --hash=sha256:da70d4d51c8b306bb7a031d5cff6cc25ad253affe89b70352af5f1cb68e74b53 \
+    --hash=sha256:daf3cb43b7cf2ba96d614252ce1684c1bccee6b2183a01328c98d36fcd7d5cb0 \
+    --hash=sha256:dca1e2f3ca00b84a396bc1bce13dd21f680f035314d2379c4160c98153b2059b \
+    --hash=sha256:dd4f49ae60e10adbc94b45c0b5e6a179acc1736cf7a90160b404076ee283cf83 \
+    --hash=sha256:e1f145462f1fa6e4a4ae3c0f782e580ce44d57c8f2c7aae1b6fa88c0b2efdb41 \
+    --hash=sha256:e3391d1e16e2a5a1507d83e4a8b100f4ee626e8eca43cf2cadb543de69827c4c \
+    --hash=sha256:fcd2469d6a2cf298f198f0487e0a5b1a47a42ca0fa4dfd1b6862c999f018ebbf \
+    --hash=sha256:fd096eb7ffef17c456cfa587523c5f92321ae02427ff955bebe9e3c63bc9f0da \
+    --hash=sha256:fe754d231288e1e64323cfad462fcee8f0288654c10bdf4f603a39ed923bef33
+    # via sqlalchemy
+grpcio==1.65.2 \
+    --hash=sha256:01600b1b02fdc9d648630d3de0a4cbf7ebe5f94b40ec1f65e3fd4b94a3b052cf \
+    --hash=sha256:03fdd86ff7d9957b822b9bf1fe0ae1e21e258e9c1d5535a5e9c67de0ad45b6a8 \
+    --hash=sha256:06a7ea12a81e5e2fb17528556c7f828b90bd2aec3a645f5cd5f35f80aa59ac6a \
+    --hash=sha256:0b2ae6868864e4b06bff89cf91730a63141327158bf0677428ef315ea1dbdb0b \
+    --hash=sha256:0c9c865d2fdf40e7e952038a0b5e0f32b01da84ecf04943b08e8917c8ccc9cf8 \
+    --hash=sha256:2a0cd7297abf0a02a9399edebe8c662058c7f0768bfbe859837707d389ad327f \
+    --hash=sha256:2ce639f2a2951aedbe9a3636f5730288f9b77c2627f116265d7d2789555e5662 \
+    --hash=sha256:3101fa25b93f185e8cc698f8c2abee897891e6bae4f13472f66df21e8ae40d46 \
+    --hash=sha256:3a3139414399078560a84203f9fe3592483d902a2af84062c571be6191143a9f \
+    --hash=sha256:3a6b36e20b02ca830b15b5eb4abb437de1d42ba93353d1f76b00337108f7ce8e \
+    --hash=sha256:3ce7f4c766fecc34455357b31b1e316506ea6ac48abbb9a650843d20337a2036 \
+    --hash=sha256:478725160e2cfc1bfa5ab3e7bb7c896cc182c8f57255d780007cfd6fb46e97b5 \
+    --hash=sha256:4fba3ae83ef5acd111c2dd92233ff167411db84e1ff17a00c34b5428355526c5 \
+    --hash=sha256:510bf7ec7f44e9420bb17970fb450522666d8b1c09cdf59b735de0c2dc806b79 \
+    --hash=sha256:51231a22aea830be1d955de5a15da4391b3ac8e1d7868f362c74c15a0e9f5c89 \
+    --hash=sha256:5225b8ce980b598187f64436ed95ea149966d538253c28668347d331968e2386 \
+    --hash=sha256:5edea0ea18e9fd5326d385a4c92a1fed605454e9a2c57ff131df0a08004b7e69 \
+    --hash=sha256:60fe2f90875f2bef105158e370fbbefadd179f8cd689bc2cee6844aca4ccb7bb \
+    --hash=sha256:67c5e5aa92b5832ae7a3399bce5b8562fb28686446732bfa17f97d5082e8501d \
+    --hash=sha256:71fa3b7a6cef62a00014205d0e707610cfd50ae54f617d296017f10c6a9fad0d \
+    --hash=sha256:76125096d2a090d4acdce0f06f9511cebe1bcfbc0bd040e495563d7a8747dda1 \
+    --hash=sha256:77fddf42bbca65ee4db679d0608e1ffa8b22b7f516c79665b7620be2f6357c85 \
+    --hash=sha256:7fd639b0988ed5114d4b2a72ea453aafcb1439dd433c61834886b92afed9c6c1 \
+    --hash=sha256:87da0fb85ba42257e450561b0264e36abe47faae07476621ae65d8f5f60f22cd \
+    --hash=sha256:8856187a359a55babfa4d49ad96f2dd7edd8be3a36b813c7a9e41ef3d763400f \
+    --hash=sha256:8886d24345bf4b1693e9c09cf6a520f0baedd2af2a876f91bb508b24d0d46041 \
+    --hash=sha256:892f03939df46d0bfcf89fe1dbcc8818f93ad6f3377587e8db6c2b1f598736c2 \
+    --hash=sha256:8aa50787bc8036bd5ea9b7ebbbd2c49c78122eb9ff98d3c217a7c146313c5030 \
+    --hash=sha256:8d6fd1206433428d0a4ba771eac70579b41a265fe835a4d8a5214c7235e69926 \
+    --hash=sha256:9b193e116e085ad4d7ef1518d79e9fedfa7688f4967f64a6246b5b196a26326a \
+    --hash=sha256:9ba9d4b3d4fc00b8083bb47a8c40a74ba3ea330713fdd59cf53c926c9a16b002 \
+    --hash=sha256:a7b752471e7ff1472ddbf3035a34fd8e24f2eac4fedbdab311e8f3e0dee889f7 \
+    --hash=sha256:aacfd499d23130578184057008ea5329732a5ac59a4fcb73c0467d86723d23c8 \
+    --hash=sha256:b10349ceebec745a47e4339ef7c4878c9b53b82ae4b0883e16544625016d6242 \
+    --hash=sha256:b6bba0f973ef6fe7434834f1b63d16bab4b50879d5bb0ca6eb0495c87d5cbc78 \
+    --hash=sha256:b7bfcbee6b32f0e4786b7813692b3907c9e444f529126b8520cac9914479b98c \
+    --hash=sha256:c2900ad06fd8f5ad8832b1ee287caccb4a957e971b2b7983e0cd7a8e7c7098fb \
+    --hash=sha256:c45977fdc675a8961875adab7f04b785f65d3fd9c737cd60b5e3a9b1392ad444 \
+    --hash=sha256:cb48342de1c3be59e6de79c6bbc01cf05562c571a3ed32f7c2e149e7934824cf \
+    --hash=sha256:cd7dc770926cc66050242eb6c63ca8ce12cd69010bf4ff7ea6e721d4f4b11e4d \
+    --hash=sha256:d388f093010a014d3b3ddf8185ff45c5279fd825d0b20e21c8076515ae61db31 \
+    --hash=sha256:e0b2bf34340999c6d938107ec2cc9bce1ea59bf08e4694cfa47e782bdbd361f4 \
+    --hash=sha256:e2c9bbb84d5517f2bccdb1836b8ee267a1757acb3cb3e575065c103220b577ac \
+    --hash=sha256:f4b7a7d68313e252e09550bd03d9d11e460dae681cf95588a131b6b3e07d1e30 \
+    --hash=sha256:f6e5a67bbf8a1b3be5535802f6e9f507d1d8d38fb32de81ec7f03706d95a9126 \
+    --hash=sha256:f931fe9b244dc23c7478c513c3ed94ded93da8bd1a95a4d97b21abdef644304a
+    # via
+    #   -r requirements.in
+    #   google-vizier
+    #   grpcio-tools
+    #   tensorboard
+    #   tensorflow
+grpcio-tools==1.62.2 \
+    --hash=sha256:04a394cf5e51ba9be412eb9f6c482b6270bd81016e033e8eb7d21b8cc28fe8b5 \
+    --hash=sha256:04c607029ae3660fb1624ed273811ffe09d57d84287d37e63b5b802a35897329 \
+    --hash=sha256:10cc3321704ecd17c93cf68c99c35467a8a97ffaaed53207e9b2da6ae0308ee1 \
+    --hash=sha256:1679b4903aed2dc5bd8cb22a452225b05dc8470a076f14fd703581efc0740cdb \
+    --hash=sha256:184b4174d4bd82089d706e8223e46c42390a6ebac191073b9772abc77308f9fa \
+    --hash=sha256:19216e1fb26dbe23d12a810517e1b3fbb8d4f98b1a3fbebeec9d93a79f092de4 \
+    --hash=sha256:19ea69e41c3565932aa28a202d1875ec56786aea46a2eab54a3b28e8a27f9517 \
+    --hash=sha256:1d768a5c07279a4c461ebf52d0cec1c6ca85c6291c71ec2703fe3c3e7e28e8c4 \
+    --hash=sha256:1fe08d2038f2b7c53259b5c49e0ad08c8e0ce2b548d8185993e7ef67e8592cca \
+    --hash=sha256:217c2ee6a7ce519a55958b8622e21804f6fdb774db08c322f4c9536c35fdce7c \
+    --hash=sha256:2ce149ea55eadb486a7fb75a20f63ef3ac065ee6a0240ed25f3549ce7954c653 \
+    --hash=sha256:2ed775e844566ce9ce089be9a81a8b928623b8ee5820f5e4d58c1a9d33dfc5ae \
+    --hash=sha256:3708a747aa4b6b505727282ca887041174e146ae030ebcadaf4c1d346858df62 \
+    --hash=sha256:3a8d6f07e64c0c7756f4e0c4781d9d5a2b9cc9cbd28f7032a6fb8d4f847d0445 \
+    --hash=sha256:3c53b221378b035ae2f1881cbc3aca42a6075a8e90e1a342c2f205eb1d1aa6a1 \
+    --hash=sha256:40cd4eeea4b25bcb6903b82930d579027d034ba944393c4751cdefd9c49e6989 \
+    --hash=sha256:45db5da2bcfa88f2b86b57ef35daaae85c60bd6754a051d35d9449c959925b57 \
+    --hash=sha256:462e0ab8dd7c7b70bfd6e3195eebc177549ede5cf3189814850c76f9a340d7ce \
+    --hash=sha256:47117c8a7e861382470d0e22d336e5a91fdc5f851d1db44fa784b9acea190d87 \
+    --hash=sha256:472505d030135d73afe4143b0873efe0dcb385bd6d847553b4f3afe07679af00 \
+    --hash=sha256:4f6f32d39283ea834a493fccf0ebe9cfddee7577bdcc27736ad4be1732a36399 \
+    --hash=sha256:4f955702dc4b530696375251319d05223b729ed24e8673c2129f7a75d2caefbb \
+    --hash=sha256:4f989e5cebead3ae92c6abf6bf7b19949e1563a776aea896ac5933f143f0c45d \
+    --hash=sha256:58cbb24b3fa6ae35aa9c210fcea3a51aa5fef0cd25618eb4fd94f746d5a9b703 \
+    --hash=sha256:5b07b5874187e170edfbd7aa2ca3a54ebf3b2952487653e8c0b0d83601c33035 \
+    --hash=sha256:5fd5e1582b678e6b941ee5f5809340be5e0724691df5299aae8226640f94e18f \
+    --hash=sha256:6413581e14a80e0b4532577766cf0586de4dd33766a31b3eb5374a746771c07d \
+    --hash=sha256:72b61332f1b439c14cbd3815174a8f1d35067a02047c32decd406b3a09bb9890 \
+    --hash=sha256:759c60f24c33a181bbbc1232a6752f9b49fbb1583312a4917e2b389fea0fb0f2 \
+    --hash=sha256:76eb459bdf3fb666e01883270beee18f3f11ed44488486b61cd210b4e0e17cc1 \
+    --hash=sha256:7a49bccae1c7d154b78e991885c3111c9ad8c8fa98e91233de425718f47c6139 \
+    --hash=sha256:7d8b4e00c3d7237b92260fc18a561cd81f1da82e8be100db1b7d816250defc66 \
+    --hash=sha256:7ea369c4d1567d1acdf69c8ea74144f4ccad9e545df7f9a4fc64c94fa7684ba3 \
+    --hash=sha256:8214820990d01b52845f9fbcb92d2b7384a0c321b303e3ac614c219dc7d1d3af \
+    --hash=sha256:8c616d0ad872e3780693fce6a3ac8ef00fc0963e6d7815ce9dcfae68ba0fc287 \
+    --hash=sha256:987e774f74296842bbffd55ea8826370f70c499e5b5f71a8cf3103838b6ee9c3 \
+    --hash=sha256:9be84ff6d47fd61462be7523b49d7ba01adf67ce4e1447eae37721ab32464dd8 \
+    --hash=sha256:9d41e0e47dd075c075bb8f103422968a65dd0d8dc8613288f573ae91eb1053ba \
+    --hash=sha256:9f1ba79a253df9e553d20319c615fa2b429684580fa042dba618d7f6649ac7e4 \
+    --hash=sha256:a34d97c62e61bfe9e6cff0410fe144ac8cca2fc979ad0be46b7edf026339d161 \
+    --hash=sha256:a7e439476b29d6dac363b321781a113794397afceeb97dad85349db5f1cb5e9a \
+    --hash=sha256:ab84bae88597133f6ea7a2bdc57b2fda98a266fe8d8d4763652cbefd20e73ad7 \
+    --hash=sha256:b6746bc823958499a3cf8963cc1de00072962fb5e629f26d658882d3f4c35095 \
+    --hash=sha256:b8574469ecc4ff41d6bb95f44e0297cdb0d95bade388552a9a444db9cd7485cd \
+    --hash=sha256:bdc5dd3f57b5368d5d661d5d3703bcaa38bceca59d25955dff66244dbc987271 \
+    --hash=sha256:c195d74fe98541178ece7a50dad2197d43991e0f77372b9a88da438be2486f12 \
+    --hash=sha256:c384c838b34d1b67068e51b5bbe49caa6aa3633acd158f1ab16b5da8d226bc53 \
+    --hash=sha256:c48fabe40b9170f4e3d7dd2c252e4f1ff395dc24e49ac15fc724b1b6f11724da \
+    --hash=sha256:cbb8453ae83a1db2452b7fe0f4b78e4a8dd32be0f2b2b73591ae620d4d784d3d \
+    --hash=sha256:d58389fe8be206ddfb4fa703db1e24c956856fcb9a81da62b13577b3a8f7fda7 \
+    --hash=sha256:d82f681c9a9d933a9d8068e8e382977768e7779ddb8870fa0cf918d8250d1532 \
+    --hash=sha256:e33b59fb3efdddeb97ded988a871710033e8638534c826567738d3edce528752 \
+    --hash=sha256:ec674b4440ef4311ac1245a709e87b36aca493ddc6850eebe0b278d1f2b6e7d1 \
+    --hash=sha256:fa107460c842e4c1a6266150881694fefd4f33baa544ea9489601810c2210ef8
+    # via google-vizier
+h5py==3.11.0 \
+    --hash=sha256:083e0329ae534a264940d6513f47f5ada617da536d8dccbafc3026aefc33c90e \
+    --hash=sha256:1625fd24ad6cfc9c1ccd44a66dac2396e7ee74940776792772819fc69f3a3731 \
+    --hash=sha256:21dbdc5343f53b2e25404673c4f00a3335aef25521bd5fa8c707ec3833934892 \
+    --hash=sha256:52c416f8eb0daae39dabe71415cb531f95dce2d81e1f61a74537a50c63b28ab3 \
+    --hash=sha256:55106b04e2c83dfb73dc8732e9abad69d83a436b5b82b773481d95d17b9685e1 \
+    --hash=sha256:67462d0669f8f5459529de179f7771bd697389fcb3faab54d63bf788599a48ea \
+    --hash=sha256:6c4b760082626120031d7902cd983d8c1f424cdba2809f1067511ef283629d4b \
+    --hash=sha256:731839240c59ba219d4cb3bc5880d438248533366f102402cfa0621b71796b62 \
+    --hash=sha256:754c0c2e373d13d6309f408325343b642eb0f40f1a6ad21779cfa9502209e150 \
+    --hash=sha256:75bd7b3d93fbeee40860fd70cdc88df4464e06b70a5ad9ce1446f5f32eb84007 \
+    --hash=sha256:77b19a40788e3e362b54af4dcf9e6fde59ca016db2c61360aa30b47c7b7cef00 \
+    --hash=sha256:7b7e8f78072a2edec87c9836f25f34203fd492a4475709a18b417a33cfb21fa9 \
+    --hash=sha256:8ec9df3dd2018904c4cc06331951e274f3f3fd091e6d6cc350aaa90fa9b42a76 \
+    --hash=sha256:a76cae64080210389a571c7d13c94a1a6cf8cb75153044fd1f822a962c97aeab \
+    --hash=sha256:aa6ae84a14103e8dc19266ef4c3e5d7c00b68f21d07f2966f0ca7bdb6c2761fb \
+    --hash=sha256:bbd732a08187a9e2a6ecf9e8af713f1d68256ee0f7c8b652a32795670fb481ba \
+    --hash=sha256:c072655ad1d5fe9ef462445d3e77a8166cbfa5e599045f8aa3c19b75315f10e5 \
+    --hash=sha256:d9c944d364688f827dc889cf83f1fca311caf4fa50b19f009d1f2b525edd33a3 \
+    --hash=sha256:ef4e2f338fc763f50a8113890f455e1a70acd42a4d083370ceb80c463d803972 \
+    --hash=sha256:f3736fe21da2b7d8a13fe8fe415f1272d2a1ccdeff4849c1421d2fb30fd533bc \
+    --hash=sha256:f4e025e852754ca833401777c25888acb96889ee2c27e7e629a19aee288833f0
+    # via
+    #   keras
+    #   tensorflow
+idna==3.7 \
+    --hash=sha256:028ff3aadf0609c1fd278d8ea3089299412a7a8b9bd005dd08b9f8285bcb5cfc \
+    --hash=sha256:82fee1fc78add43492d3a1898bfa6d8a904cc97d8427f683ed8e798d07761aa0
+    # via requests
+jax==0.4.14 \
+    --hash=sha256:18fed3881f26e8b13c8cb46eeeea3dba9eb4d48e3714d8e8f2304dd6e237083d
+    # via -r requirements.in
+jaxlib==0.4.14 \
+    --hash=sha256:137763b4321a86a84cde4f2ee6f51d74416ca8f7dd79059ebf88403f779829ba \
+    --hash=sha256:2dadf061b1828d64e5ee4ac10fa05a051ab2043f11e77d0b7246f3d005afd053 \
+    --hash=sha256:504e478acde687fd3ba8ee0b250fb675e26d92b4063d97435d4ea8c5fec360b3 \
+    --hash=sha256:5739b7af01d8765c982fd3e5c462ff3ac63d37d583114df639bdc318deb790cb \
+    --hash=sha256:6b38dae702bb8c1b4b54241bf04acefbada9813e606233e59fbc2a51395f650a \
+    --hash=sha256:843839faa8ac82d80eaddbdb8f1b14808ef7716e806a1f0d2e384f7b890c02c3 \
+    --hash=sha256:9f2ca54997ea1de7ed96f56025914237595d8e5b99549218bb8652b15c6f5dff \
+    --hash=sha256:aacea888b9de76ea3b001f0952ba7ae88a4e40a7ede302344b305bbc8d2f5cde \
+    --hash=sha256:b03714c869fb78f1166a8a0371a160dd27fc047bee9cfea39c93ad64a0496e35 \
+    --hash=sha256:b934e1dbbe1adf49bdf080197db6681c59c80ad07af5e36968dd69825e79a40e \
+    --hash=sha256:fb06229143c586bbd05bc932d76aa78c83895c03a1670e267da39f7bc61223e9 \
+    --hash=sha256:fbd3fb60b3a11585e2db58c340f70a1b51a4a37fadc0c77fbe819677ed75ee78
+    # via -r requirements.in
+joblib==1.4.2 \
+    --hash=sha256:06d478d5674cbc267e7496a410ee875abd68e4340feff4490bcb7afb88060ae6 \
+    --hash=sha256:2382c5816b2636fbd20a09e0f4e9dad4736765fdfb7dca582943b9c1366b3f0e
+    # via scikit-learn
+keras==3.4.1 \
+    --hash=sha256:15599c51e2090c12f39de6db6489a0cf265ddf6653f0731b82db5af2bfa19105 \
+    --hash=sha256:34cd9aeaa008914715149234c215657ca758e1b473bd2aab2e211ac967d1f8fe
+    # via tensorflow
+libclang==18.1.1 \
+    --hash=sha256:0b2e143f0fac830156feb56f9231ff8338c20aecfe72b4ffe96f19e5a1dbb69a \
+    --hash=sha256:3f0e1f49f04d3cd198985fea0511576b0aee16f9ff0e0f0cad7f9c57ec3c20e8 \
+    --hash=sha256:4dd2d3b82fab35e2bf9ca717d7b63ac990a3519c7e312f19fa8e86dcc712f7fb \
+    --hash=sha256:54dda940a4a0491a9d1532bf071ea3ef26e6dbaf03b5000ed94dd7174e8f9592 \
+    --hash=sha256:69f8eb8f65c279e765ffd28aaa7e9e364c776c17618af8bff22a8df58677ff4f \
+    --hash=sha256:6f14c3f194704e5d09769108f03185fce7acaf1d1ae4bbb2f30a72c2400cb7c5 \
+    --hash=sha256:83ce5045d101b669ac38e6da8e58765f12da2d3aafb3b9b98d88b286a60964d8 \
+    --hash=sha256:a1214966d08d73d971287fc3ead8dfaf82eb07fb197680d8b3859dbbbbf78250 \
+    --hash=sha256:c533091d8a3bbf7460a00cb6c1a71da93bffe148f172c7d03b1c31fbf8aa2a0b \
+    --hash=sha256:cf4a99b05376513717ab5d82a0db832c56ccea4fd61a69dbb7bccf2dfb207dbe
+    # via tensorflow
+markdown==3.6 \
+    --hash=sha256:48f276f4d8cfb8ce6527c8f79e2ee29708508bf4d40aa410fbc3b4ee832c850f \
+    --hash=sha256:ed4f41f6daecbeeb96e576ce414c41d2d876daa9a16cb35fa8ed8c2ddfad0224
+    # via tensorboard
+markdown-it-py==3.0.0 \
+    --hash=sha256:355216845c60bd96232cd8d8c40e8f9765cc86f46880e43a8fd22dc1a1a8cab1 \
+    --hash=sha256:e3f60a94fa066dc52ec76661e37c851cb232d92f9886b15cb560aaada2df8feb
+    # via rich
+markupsafe==2.1.5 \
+    --hash=sha256:00e046b6dd71aa03a41079792f8473dc494d564611a8f89bbbd7cb93295ebdcf \
+    --hash=sha256:075202fa5b72c86ad32dc7d0b56024ebdbcf2048c0ba09f1cde31bfdd57bcfff \
+    --hash=sha256:0e397ac966fdf721b2c528cf028494e86172b4feba51d65f81ffd65c63798f3f \
+    --hash=sha256:17b950fccb810b3293638215058e432159d2b71005c74371d784862b7e4683f3 \
+    --hash=sha256:1f3fbcb7ef1f16e48246f704ab79d79da8a46891e2da03f8783a5b6fa41a9532 \
+    --hash=sha256:2174c595a0d73a3080ca3257b40096db99799265e1c27cc5a610743acd86d62f \
+    --hash=sha256:2b7c57a4dfc4f16f7142221afe5ba4e093e09e728ca65c51f5620c9aaeb9a617 \
+    --hash=sha256:2d2d793e36e230fd32babe143b04cec8a8b3eb8a3122d2aceb4a371e6b09b8df \
+    --hash=sha256:30b600cf0a7ac9234b2638fbc0fb6158ba5bdcdf46aeb631ead21248b9affbc4 \
+    --hash=sha256:397081c1a0bfb5124355710fe79478cdbeb39626492b15d399526ae53422b906 \
+    --hash=sha256:3a57fdd7ce31c7ff06cdfbf31dafa96cc533c21e443d57f5b1ecc6cdc668ec7f \
+    --hash=sha256:3c6b973f22eb18a789b1460b4b91bf04ae3f0c4234a0a6aa6b0a92f6f7b951d4 \
+    --hash=sha256:3e53af139f8579a6d5f7b76549125f0d94d7e630761a2111bc431fd820e163b8 \
+    --hash=sha256:4096e9de5c6fdf43fb4f04c26fb114f61ef0bf2e5604b6ee3019d51b69e8c371 \
+    --hash=sha256:4275d846e41ecefa46e2015117a9f491e57a71ddd59bbead77e904dc02b1bed2 \
+    --hash=sha256:4c31f53cdae6ecfa91a77820e8b151dba54ab528ba65dfd235c80b086d68a465 \
+    --hash=sha256:4f11aa001c540f62c6166c7726f71f7573b52c68c31f014c25cc7901deea0b52 \
+    --hash=sha256:5049256f536511ee3f7e1b3f87d1d1209d327e818e6ae1365e8653d7e3abb6a6 \
+    --hash=sha256:58c98fee265677f63a4385256a6d7683ab1832f3ddd1e66fe948d5880c21a169 \
+    --hash=sha256:598e3276b64aff0e7b3451b72e94fa3c238d452e7ddcd893c3ab324717456bad \
+    --hash=sha256:5b7b716f97b52c5a14bffdf688f971b2d5ef4029127f1ad7a513973cfd818df2 \
+    --hash=sha256:5dedb4db619ba5a2787a94d877bc8ffc0566f92a01c0ef214865e54ecc9ee5e0 \
+    --hash=sha256:619bc166c4f2de5caa5a633b8b7326fbe98e0ccbfacabd87268a2b15ff73a029 \
+    --hash=sha256:629ddd2ca402ae6dbedfceeba9c46d5f7b2a61d9749597d4307f943ef198fc1f \
+    --hash=sha256:656f7526c69fac7f600bd1f400991cc282b417d17539a1b228617081106feb4a \
+    --hash=sha256:6ec585f69cec0aa07d945b20805be741395e28ac1627333b1c5b0105962ffced \
+    --hash=sha256:72b6be590cc35924b02c78ef34b467da4ba07e4e0f0454a2c5907f473fc50ce5 \
+    --hash=sha256:7502934a33b54030eaf1194c21c692a534196063db72176b0c4028e140f8f32c \
+    --hash=sha256:7a68b554d356a91cce1236aa7682dc01df0edba8d043fd1ce607c49dd3c1edcf \
+    --hash=sha256:7b2e5a267c855eea6b4283940daa6e88a285f5f2a67f2220203786dfa59b37e9 \
+    --hash=sha256:823b65d8706e32ad2df51ed89496147a42a2a6e01c13cfb6ffb8b1e92bc910bb \
+    --hash=sha256:8590b4ae07a35970728874632fed7bd57b26b0102df2d2b233b6d9d82f6c62ad \
+    --hash=sha256:8dd717634f5a044f860435c1d8c16a270ddf0ef8588d4887037c5028b859b0c3 \
+    --hash=sha256:8dec4936e9c3100156f8a2dc89c4b88d5c435175ff03413b443469c7c8c5f4d1 \
+    --hash=sha256:97cafb1f3cbcd3fd2b6fbfb99ae11cdb14deea0736fc2b0952ee177f2b813a46 \
+    --hash=sha256:a17a92de5231666cfbe003f0e4b9b3a7ae3afb1ec2845aadc2bacc93ff85febc \
+    --hash=sha256:a549b9c31bec33820e885335b451286e2969a2d9e24879f83fe904a5ce59d70a \
+    --hash=sha256:ac07bad82163452a6884fe8fa0963fb98c2346ba78d779ec06bd7a6262132aee \
+    --hash=sha256:ae2ad8ae6ebee9d2d94b17fb62763125f3f374c25618198f40cbb8b525411900 \
+    --hash=sha256:b91c037585eba9095565a3556f611e3cbfaa42ca1e865f7b8015fe5c7336d5a5 \
+    --hash=sha256:bc1667f8b83f48511b94671e0e441401371dfd0f0a795c7daa4a3cd1dde55bea \
+    --hash=sha256:bec0a414d016ac1a18862a519e54b2fd0fc8bbfd6890376898a6c0891dd82e9f \
+    --hash=sha256:bf50cd79a75d181c9181df03572cdce0fbb75cc353bc350712073108cba98de5 \
+    --hash=sha256:bff1b4290a66b490a2f4719358c0cdcd9bafb6b8f061e45c7a2460866bf50c2e \
+    --hash=sha256:c061bb86a71b42465156a3ee7bd58c8c2ceacdbeb95d05a99893e08b8467359a \
+    --hash=sha256:c8b29db45f8fe46ad280a7294f5c3ec36dbac9491f2d1c17345be8e69cc5928f \
+    --hash=sha256:ce409136744f6521e39fd8e2a24c53fa18ad67aa5bc7c2cf83645cce5b5c4e50 \
+    --hash=sha256:d050b3361367a06d752db6ead6e7edeb0009be66bc3bae0ee9d97fb326badc2a \
+    --hash=sha256:d283d37a890ba4c1ae73ffadf8046435c76e7bc2247bbb63c00bd1a709c6544b \
+    --hash=sha256:d9fad5155d72433c921b782e58892377c44bd6252b5af2f67f16b194987338a4 \
+    --hash=sha256:daa4ee5a243f0f20d528d939d06670a298dd39b1ad5f8a72a4275124a7819eff \
+    --hash=sha256:db0b55e0f3cc0be60c1f19efdde9a637c32740486004f20d1cff53c3c0ece4d2 \
+    --hash=sha256:e61659ba32cf2cf1481e575d0462554625196a1f2fc06a1c777d3f48e8865d46 \
+    --hash=sha256:ea3d8a3d18833cf4304cd2fc9cbb1efe188ca9b5efef2bdac7adc20594a0e46b \
+    --hash=sha256:ec6a563cff360b50eed26f13adc43e61bc0c04d94b8be985e6fb24b81f6dcfdf \
+    --hash=sha256:f5dfb42c4604dddc8e4305050aa6deb084540643ed5804d7455b5df8fe16f5e5 \
+    --hash=sha256:fa173ec60341d6bb97a89f5ea19c85c5643c1e7dedebc22f5181eb73573142c5 \
+    --hash=sha256:fa9db3f79de01457b03d4f01b34cf91bc0048eb2c3846ff26f66687c2f6d16ab \
+    --hash=sha256:fce659a462a1be54d2ffcacea5e3ba2d74daa74f30f5f143fe0c58636e355fdd \
+    --hash=sha256:ffee1f21e5ef0d712f9033568f8344d5da8cc2869dbd08d87c84656e6a2d2f68
+    # via werkzeug
+mdurl==0.1.2 \
+    --hash=sha256:84008a41e51615a49fc9966191ff91509e3c40b939176e643fd50a5c2196b8f8 \
+    --hash=sha256:bb413d29f5eea38f31dd4754dd7377d4465116fb207585f97bf925588687c1ba
+    # via markdown-it-py
+ml-dtypes==0.3.2 \
+    --hash=sha256:2c34f2ba9660b21fe1034b608308a01be82bbef2a92fb8199f24dc6bad0d5226 \
+    --hash=sha256:3a17ef2322e60858d93584e9c52a5be7dd6236b056b7fa1ec57f1bb6ba043e33 \
+    --hash=sha256:533059bc5f1764fac071ef54598db358c167c51a718f68f5bb55e3dee79d2967 \
+    --hash=sha256:6604877d567a29bfe7cc02969ae0f2425260e5335505cf5e7fefc3e5465f5655 \
+    --hash=sha256:6b35c4e8ca957c877ac35c79ffa77724ecc3702a1e4b18b08306c03feae597bb \
+    --hash=sha256:763697ab8a88d47443997a7cdf3aac7340049aed45f7521f6b0ec8a0594821fe \
+    --hash=sha256:7a4c3fcbf86fa52d0204f07cfd23947ef05b4ad743a1a988e163caa34a201e5e \
+    --hash=sha256:7afde548890a92b41c0fed3a6c525f1200a5727205f73dc21181a2726571bb53 \
+    --hash=sha256:7ba8e1fafc7fff3e643f453bffa7d082df1678a73286ce8187d3e825e776eb94 \
+    --hash=sha256:91f8783fd1f2c23fd3b9ee5ad66b785dafa58ba3cdb050c4458021fa4d1eb226 \
+    --hash=sha256:93b78f53431c93953f7850bb1b925a17f0ab5d97527e38a7e865b5b4bc5cfc18 \
+    --hash=sha256:961134ea44c7b8ca63eda902a44b58cd8bd670e21d62e255c81fba0a8e70d9b7 \
+    --hash=sha256:b89b194e9501a92d289c1ffd411380baf5daafb9818109a4f49b0a1b6dce4462 \
+    --hash=sha256:c7b3fb3d4f6b39bcd4f6c4b98f406291f0d681a895490ee29a0f95bab850d53c \
+    --hash=sha256:d1a746fe5fb9cd974a91070174258f0be129c592b93f9ce7df6cc336416c3fbd \
+    --hash=sha256:e8505946df1665db01332d885c2020b4cb9e84a8b1241eb4ba69d59591f65855 \
+    --hash=sha256:f47619d978ab1ae7dfdc4052ea97c636c6263e1f19bd1be0e42c346b98d15ff4
+    # via
+    #   jax
+    #   jaxlib
+    #   keras
+    #   tensorflow
+mpmath==1.3.0 \
+    --hash=sha256:7a28eb2a9774d00c7bc92411c19a89209d5da7c4c9a9e227be8330a23a25b91f \
+    --hash=sha256:a0b2b9fe80bbcd81a6647ff13108738cfb482d481d826cc0e02f5b35e5c88d2c
+    # via dp-accounting
+namex==0.0.8 \
+    --hash=sha256:32a50f6c565c0bb10aa76298c959507abdc0e850efe085dc38f3440fcb3aa90b \
+    --hash=sha256:7ddb6c2bb0e753a311b7590f84f6da659dd0c05e65cb89d519d54c0a250c0487
+    # via keras
+numpy==1.25.2 \
+    --hash=sha256:0d60fbae8e0019865fc4784745814cff1c421df5afee233db6d88ab4f14655a2 \
+    --hash=sha256:1a1329e26f46230bf77b02cc19e900db9b52f398d6722ca853349a782d4cff55 \
+    --hash=sha256:1b9735c27cea5d995496f46a8b1cd7b408b3f34b6d50459d9ac8fe3a20cc17bf \
+    --hash=sha256:2792d23d62ec51e50ce4d4b7d73de8f67a2fd3ea710dcbc8563a51a03fb07b01 \
+    --hash=sha256:3e0746410e73384e70d286f93abf2520035250aad8c5714240b0492a7302fdca \
+    --hash=sha256:4c3abc71e8b6edba80a01a52e66d83c5d14433cbcd26a40c329ec7ed09f37901 \
+    --hash=sha256:5883c06bb92f2e6c8181df7b39971a5fb436288db58b5a1c3967702d4278691d \
+    --hash=sha256:5c97325a0ba6f9d041feb9390924614b60b99209a71a69c876f71052521d42a4 \
+    --hash=sha256:60e7f0f7f6d0eee8364b9a6304c2845b9c491ac706048c7e8cf47b83123b8dbf \
+    --hash=sha256:76b4115d42a7dfc5d485d358728cdd8719be33cc5ec6ec08632a5d6fca2ed380 \
+    --hash=sha256:7dc869c0c75988e1c693d0e2d5b26034644399dd929bc049db55395b1379e044 \
+    --hash=sha256:834b386f2b8210dca38c71a6e0f4fd6922f7d3fcff935dbe3a570945acb1b545 \
+    --hash=sha256:8b77775f4b7df768967a7c8b3567e309f617dd5e99aeb886fa14dc1a0791141f \
+    --hash=sha256:90319e4f002795ccfc9050110bbbaa16c944b1c37c0baeea43c5fb881693ae1f \
+    --hash=sha256:b79e513d7aac42ae918db3ad1341a015488530d0bb2a6abcbdd10a3a829ccfd3 \
+    --hash=sha256:bb33d5a1cf360304754913a350edda36d5b8c5331a8237268c48f91253c3a364 \
+    --hash=sha256:bec1e7213c7cb00d67093247f8c4db156fd03075f49876957dca4711306d39c9 \
+    --hash=sha256:c5462d19336db4560041517dbb7759c21d181a67cb01b36ca109b2ae37d32418 \
+    --hash=sha256:c5652ea24d33585ea39eb6a6a15dac87a1206a692719ff45d53c5282e66d4a8f \
+    --hash=sha256:d7806500e4f5bdd04095e849265e55de20d8cc4b661b038957354327f6d9b295 \
+    --hash=sha256:db3ccc4e37a6873045580d413fe79b68e47a681af8db2e046f1dacfa11f86eb3 \
+    --hash=sha256:dfe4a913e29b418d096e696ddd422d8a5d13ffba4ea91f9f60440a3b759b0187 \
+    --hash=sha256:eb942bfb6f84df5ce05dbf4b46673ffed0d3da59f13635ea9b926af3deb76926 \
+    --hash=sha256:f08f2e037bba04e707eebf4bc934f1972a315c883a9e0ebfa8a7756eabf9e357 \
+    --hash=sha256:fd608e19c8d7c55021dffd43bfe5492fab8cc105cc8986f813f8c3c048b38760
+    # via
+    #   -r requirements.in
+    #   dp-accounting
+    #   google-vizier
+    #   h5py
+    #   jax
+    #   jaxlib
+    #   keras
+    #   ml-dtypes
+    #   opt-einsum
+    #   scikit-learn
+    #   scipy
+    #   tensorboard
+    #   tensorflow
+    #   tensorflow-model-optimization
+    #   tensorflow-privacy
+    #   tensorflow-probability
+opt-einsum==3.3.0 \
+    --hash=sha256:2455e59e3947d3c275477df7f5205b30635e266fe6dc300e3d9f9646bfcea147 \
+    --hash=sha256:59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549
+    # via
+    #   jax
+    #   tensorflow
+optree==0.12.1 \
+    --hash=sha256:06d6ef39b3ef9920d6cdb6d3d1d2804a37092d24dc406c4cb9b46cd6c9a44e89 \
+    --hash=sha256:154738def491199d3fbcd919437315728e0a1caeaf4ec06688c76ef9d56e5ed6 \
+    --hash=sha256:1b2fe5c04c218698a53ed2d4b7372f1989df8cf0a61d616e6f384770d8a5fb1c \
+    --hash=sha256:1d76905bced5cf569d23dc4890341fae2fa257cce58a492a1603afcdc5969ae7 \
+    --hash=sha256:1f8baf0ad6b58843d24fa8caf079cf1f0c33cc3658263cff960b5c1d0cc53bc8 \
+    --hash=sha256:23afe4aae42336bdf8cf4fba35c56593405bf8f8e163627f722205b3bf0d9310 \
+    --hash=sha256:24d74a9d97d7bdbdbb30356850f204950c39ab8fad7f273ed29d1feda19060b2 \
+    --hash=sha256:27ae426745931ae1c2ccd7a78b27f9b7402167e0600fa62e2ef1cd58727e7b94 \
+    --hash=sha256:2a1a9905d2d917d5aff775283e0a59be2c6b529a219241c248d50b3ad51c6cce \
+    --hash=sha256:2d4d8e024b841f99907b2340fee7ac9994fbe300383a9af6c93578d12861a969 \
+    --hash=sha256:2de1297b2bf019379ab86103e31caa97c8a08628f0c8b58cd7709f9048c589eb \
+    --hash=sha256:349aafac463642979f7fe7ca3aa9e2fa8a5a0f81ef7af6946a075b797673e600 \
+    --hash=sha256:35ca77b810cf5959e6930d56534ecbecc4300f5e5fa14b977030265c1c8eab6c \
+    --hash=sha256:3e323744d083bd8b4648c9ff2383f01bfbc33098656d56fdd984b2263ef905f3 \
+    --hash=sha256:404cf2decd8fb6a1a8f6fef623c98873cdf7ae086aeb8909d104cd321d829ba0 \
+    --hash=sha256:409ef6f3656299923d722509849d83607bb3e5c621dcfe6aa90ace85665e9b54 \
+    --hash=sha256:411a21eca034ddb98eb80e6c4bf552fc46b8d8ab7c4d250446d74d31a251a684 \
+    --hash=sha256:42025da0bac19cc6de756fe64511f15baffb3fa7d8402e54aab035c02903eb5c \
+    --hash=sha256:47db001a224382493ae7a8df16e7a9668e971fc129970d137995421aa6b06f8f \
+    --hash=sha256:4b32f39988bfe6e76eeefb335da529e614145f7f1dfa8583fbc4aca8a72f504b \
+    --hash=sha256:4ee926120887404e92877c99714b960bc29f572e8db69fd2e934022d80452f91 \
+    --hash=sha256:50893bd088bdb3e2f07ee481dafd848b483bea1a19cc978f2309139314e5bc7d \
+    --hash=sha256:509bddd38dae8c4e8d6b988f514b7a9fe803ca916b11af67b40520f0b1eeeaef \
+    --hash=sha256:562036d3de15204ed1a88d9fc262a7e1c20964d22ef132069e20dbd88215f983 \
+    --hash=sha256:5bfe3d3e47e10b528f9324d446c871bfad7d0be8c2bd2a2fbc3ddf1600ae8558 \
+    --hash=sha256:5c2f2e0e3978558bc8f7df8c5a999674097dd0dc71363210783eb8d7a6da8ef9 \
+    --hash=sha256:5f24b0a8b181a90a778cadc942a79336d29f0c164704d58cd20989bf7d0bea1c \
+    --hash=sha256:606983f4696d81128e205a1c34d0c9f3fe6ae12f6c26ed5e8ab3722d6f378ec2 \
+    --hash=sha256:62d232a344c14b8e94fdd6de1acf2c0b05954b05d6bb346bddb13c38be37dc09 \
+    --hash=sha256:646842f8a2de2caaacc32a8c91f8031a93eda145ac9c915bb0fd2ad5249c14b7 \
+    --hash=sha256:6d90fb28d52725352858013cafe34d98d90ab1bb86b5d8dc29d420e9bbc5706b \
+    --hash=sha256:76a2240e7482355966a73c6c701e3d1f148420a77849c78d175d3b08bf06ff36 \
+    --hash=sha256:7a71dd58522cd6258b61b639092ac7a2631d881f039ef968b31dfd555e513591 \
+    --hash=sha256:7e79eedd9406c59d542482768e490795dc6b6f1a014c7852d29d9fd61749bf94 \
+    --hash=sha256:80e0d4eba4a65d4c6f2002ed949142a40933b8185523894659c26c34693c4086 \
+    --hash=sha256:8513d6dd71807abb1037a5b5bc66b45c21afb42e9c90961fa5e762cea3943ab2 \
+    --hash=sha256:88d01ce6f78f209cad8dc4cf2d3222d7056cac93612abfd6beb40ab43a131769 \
+    --hash=sha256:9280452c11da0872ec57be5d8f153207d6303b3cbf26115b2bf6d2b8157a5343 \
+    --hash=sha256:9c473988b2d8fd7edc3958e6c7cb1d3f92afb7bcaff53b76a8f41cf4f3a24709 \
+    --hash=sha256:a11e58d7c0a71a48d74ca0a6715f4c0932c6f9409ba93d600e3326df4cf778ae \
+    --hash=sha256:a49d3cfec1a51463b63e11c889bb00207c4e040016833cd202871ad946116925 \
+    --hash=sha256:a55a79c1c72f73259532e4cbe9ff65bed9663064747db02591fb4714fe742d2e \
+    --hash=sha256:a67842cd1c5c83d74863872f06fe6ed64e44279c0378267a9805567fe3c38591 \
+    --hash=sha256:aadb26d68f1d7871507f84846d8844aa94f47402d5277ce19cfe5102bb5df9e9 \
+    --hash=sha256:afa0051335c6032ee4dfc212952dcfb3b23fe59bcd70f56d25a214e7585cd62c \
+    --hash=sha256:b1ca00bdfe4da8068c2773b7ac4c8c96d3f61b8d21eba6a8642dab23ee631b0d \
+    --hash=sha256:b43c09cf9dd28aed2efc163f4bb4808d7fad54250812960bf349399ba6972e16 \
+    --hash=sha256:b890ba0a21049addf589c314c85e98a68d3dfc84e3954491e9ce60f60cb7b0e7 \
+    --hash=sha256:ba6aed8b9684c5804a5e2d6b246c3b4a68bab793b6829d369ba1c53734852a0c \
+    --hash=sha256:bd207b43e71fb3f8c315e2e4a5444f48317b2108889e96279d5426bca730a47e \
+    --hash=sha256:c987931bd31d0f28cbff28925a875639170534a36ce178a40020aca0769d9549 \
+    --hash=sha256:ce7cb233e87a2dc127b8ec82bd61f098e6ff1e57d0a09dc110a17b38bfd73034 \
+    --hash=sha256:cefd4f4c7596cdd4c95dca431bc41284a43ebd7056e739480f157789aa34579d \
+    --hash=sha256:d0950ee245db2c40824362def1efc15621a6492419628cec1fac0061818420f7 \
+    --hash=sha256:d313303a1ce36ea55c3a96fc375c5cc64a9ab814ab2677ce64e4a7d755a9b1d0 \
+    --hash=sha256:d913122454d0e3f10dc25a1b598eaf588d225372f41ece3ad4d508bddd363e4d \
+    --hash=sha256:da37e6cc669a9840844722edb3f8dd5b4f07e99b0e8c9196089cb49af70c7b75 \
+    --hash=sha256:e124f30daf79d51b1bbbda7e74d01e637fa47aff4aa64cb082b88057535daa64 \
+    --hash=sha256:e2027217c3acaf44e5f5aabe01ba0cbf33066f3f6df870881ddf597965f80db0 \
+    --hash=sha256:e20b5569369a5f1e8faa2604799b91a1941fe17b5de8afc84c8c23ff66d8e585 \
+    --hash=sha256:e8046cbbcd5f7494ba7c6811e44a6d2867216f2bdb7cef980a9a62e31d39270c \
+    --hash=sha256:eb968d3cc1db8944f220f1a67c9db043b86b47ace90ce3cfd23f3e6500baeb65 \
+    --hash=sha256:efffa3814ab8e3aaf7bf88495e4b6d263de9689d6f02dfa4490f8f64736806ac \
+    --hash=sha256:f0460f025bf1c08f2c008b5e3628d849fcb5810345222e57879cd248fec7f9f7 \
+    --hash=sha256:f65a31d7cfab2fed2bc29ab6eabcf4205dec6e0ee3cfb7006336c4f76d78fb0e \
+    --hash=sha256:f6b98b80b1259e9817aca701beba616ce33e43e856e7d644f7e0f582b8e45565 \
+    --hash=sha256:fc1ec38d1ec43bb8358ab058c3220a70b7bfb56f2bb625f41cb09d117a0d6150 \
+    --hash=sha256:fd3ead0c64d22d692284d96c27d5091e682b002ffe5a52afacc9f1fcc8ae3180
+    # via keras
+packaging==22.0 \
+    --hash=sha256:2198ec20bd4c017b8f9717e00f0c8714076fc2fd93816750ab48e2c41de2cfd3 \
+    --hash=sha256:957e2148ba0e1a3b282772e791ef1d8083648bc131c8ab0c1feba110ce1146c3
+    # via
+    #   keras
+    #   tensorflow
+    #   tensorflow-privacy
+portpicker==1.6.0 \
+    --hash=sha256:b2787a41404cf7edbe29b07b9e0ed863b09f2665dcc01c1eb0c2261c1e7d0755 \
+    --hash=sha256:bd507fd6f96f65ee02781f2e674e9dc6c99bbfa6e3c39992e3916204c9d431fa
+    # via
+    #   -r requirements.in
+    #   google-vizier
+protobuf==4.25.4 \
+    --hash=sha256:051e97ce9fa6067a4546e75cb14f90cf0232dcb3e3d508c448b8d0e4265b61c1 \
+    --hash=sha256:0dc4a62cc4052a036ee2204d26fe4d835c62827c855c8a03f29fe6da146b380d \
+    --hash=sha256:3319e073562e2515c6ddc643eb92ce20809f5d8f10fead3332f71c63be6a7040 \
+    --hash=sha256:4c8a70fdcb995dcf6c8966cfa3a29101916f7225e9afe3ced4395359955d3835 \
+    --hash=sha256:7e372cbbda66a63ebca18f8ffaa6948455dfecc4e9c1029312f6c2edcd86c4e1 \
+    --hash=sha256:90bf6fd378494eb698805bbbe7afe6c5d12c8e17fca817a646cd6a1818c696ca \
+    --hash=sha256:ac79a48d6b99dfed2729ccccee547b34a1d3d63289c71cef056653a846a2240f \
+    --hash=sha256:ba3d8504116a921af46499471c63a85260c1a5fc23333154a427a310e015d26d \
+    --hash=sha256:bfbebc1c8e4793cfd58589acfb8a1026be0003e852b9da7db5a4285bde996978 \
+    --hash=sha256:db9fd45183e1a67722cafa5c1da3e85c6492a5383f127c86c4c4aa4845867dc4 \
+    --hash=sha256:eecd41bfc0e4b1bd3fa7909ed93dd14dd5567b98c941d6c1ad08fdcab3d6884b
+    # via
+    #   google-vizier
+    #   googleapis-common-protos
+    #   grpcio-tools
+    #   tensorboard
+    #   tensorflow
+psutil==6.0.0 \
+    --hash=sha256:02b69001f44cc73c1c5279d02b30a817e339ceb258ad75997325e0e6169d8b35 \
+    --hash=sha256:1287c2b95f1c0a364d23bc6f2ea2365a8d4d9b726a3be7294296ff7ba97c17f0 \
+    --hash=sha256:1e7c870afcb7d91fdea2b37c24aeb08f98b6d67257a5cb0a8bc3ac68d0f1a68c \
+    --hash=sha256:21f1fb635deccd510f69f485b87433460a603919b45e2a324ad65b0cc74f8fb1 \
+    --hash=sha256:33ea5e1c975250a720b3a6609c490db40dae5d83a4eb315170c4fe0d8b1f34b3 \
+    --hash=sha256:34859b8d8f423b86e4385ff3665d3f4d94be3cdf48221fbe476e883514fdb71c \
+    --hash=sha256:5fd9a97c8e94059b0ef54a7d4baf13b405011176c3b6ff257c247cae0d560ecd \
+    --hash=sha256:6ec7588fb3ddaec7344a825afe298db83fe01bfaaab39155fa84cf1c0d6b13c3 \
+    --hash=sha256:6ed2440ada7ef7d0d608f20ad89a04ec47d2d3ab7190896cd62ca5fc4fe08bf0 \
+    --hash=sha256:8faae4f310b6d969fa26ca0545338b21f73c6b15db7c4a8d934a5482faa818f2 \
+    --hash=sha256:a021da3e881cd935e64a3d0a20983bda0bb4cf80e4f74fa9bfcb1bc5785360c6 \
+    --hash=sha256:a495580d6bae27291324fe60cea0b5a7c23fa36a7cd35035a16d93bdcf076b9d \
+    --hash=sha256:a9a3dbfb4de4f18174528d87cc352d1f788b7496991cca33c6996f40c9e3c92c \
+    --hash=sha256:c588a7e9b1173b6e866756dde596fd4cad94f9399daf99ad8c3258b3cb2b47a0 \
+    --hash=sha256:e2e8d0054fc88153ca0544f5c4d554d42e33df2e009c4ff42284ac9ebdef4132 \
+    --hash=sha256:fc8c9510cde0146432bbdb433322861ee8c3efbf8589865c8bf8d21cb30c4d14 \
+    --hash=sha256:ffe7fc9b6b36beadc8c322f84e1caff51e8703b88eee1da46d1e3a6ae11b4fd0
+    # via portpicker
+pygments==2.18.0 \
+    --hash=sha256:786ff802f32e91311bff3889f6e9a86e81505fe99f2735bb6d60ae0c5004f199 \
+    --hash=sha256:b8e6aca0523f3ab76fee51799c488e38782ac06eafcf95e7ba832985c8e7b13a
+    # via rich
+requests==2.32.3 \
+    --hash=sha256:55365417734eb18255590a9ff9eb97e9e1da868d4ccd6402399eaf68af20a760 \
+    --hash=sha256:70761cfe03c773ceb22aa2f671b4757976145175cdfca038c02654d061d6dcc6
+    # via tensorflow
+rich==13.7.1 \
+    --hash=sha256:4edbae314f59eb482f54e9e30bf00d33350aaa94f4bfcd4e9e3110e64d0d7222 \
+    --hash=sha256:9be308cb1fe2f1f57d67ce99e95af38a1e2bc71ad9813b0e247cf7ffbcc3a432
+    # via keras
+scikit-learn==1.5.1 \
+    --hash=sha256:0828673c5b520e879f2af6a9e99eee0eefea69a2188be1ca68a6121b809055c1 \
+    --hash=sha256:0ea5d40c0e3951df445721927448755d3fe1d80833b0b7308ebff5d2a45e6414 \
+    --hash=sha256:10e49170691514a94bb2e03787aa921b82dbc507a4ea1f20fd95557862c98dc1 \
+    --hash=sha256:154297ee43c0b83af12464adeab378dee2d0a700ccd03979e2b821e7dd7cc1c2 \
+    --hash=sha256:161808750c267b77b4a9603cf9c93579c7a74ba8486b1336034c2f1579546d21 \
+    --hash=sha256:1bd8d3a19d4bd6dc5a7d4f358c8c3a60934dc058f363c34c0ac1e9e12a31421d \
+    --hash=sha256:1ff4ba34c2abff5ec59c803ed1d97d61b036f659a17f55be102679e88f926fac \
+    --hash=sha256:508907e5f81390e16d754e8815f7497e52139162fd69c4fdbd2dfa5d6cc88915 \
+    --hash=sha256:5944ce1faada31c55fb2ba20a5346b88e36811aab504ccafb9f0339e9f780395 \
+    --hash=sha256:5f57428de0c900a98389c4a433d4a3cf89de979b3aa24d1c1d251802aa15e44d \
+    --hash=sha256:689b6f74b2c880276e365fe84fe4f1befd6a774f016339c65655eaff12e10cbf \
+    --hash=sha256:781586c414f8cc58e71da4f3d7af311e0505a683e112f2f62919e3019abd3745 \
+    --hash=sha256:7b073a27797a283187a4ef4ee149959defc350b46cbf63a84d8514fe16b69855 \
+    --hash=sha256:88e0672c7ac21eb149d409c74cc29f1d611d5158175846e7a9c2427bd12b3956 \
+    --hash=sha256:909144d50f367a513cee6090873ae582dba019cb3fca063b38054fa42704c3a4 \
+    --hash=sha256:97625f217c5c0c5d0505fa2af28ae424bd37949bb2f16ace3ff5f2f81fb4498b \
+    --hash=sha256:9a07f90846313a7639af6a019d849ff72baadfa4c74c778821ae0fad07b7275b \
+    --hash=sha256:b59e3e62d2be870e5c74af4e793293753565c7383ae82943b83383fdcf5cc5c1 \
+    --hash=sha256:b5e865e9bd59396220de49cb4a57b17016256637c61b4c5cc81aaf16bc123bbe \
+    --hash=sha256:da3f404e9e284d2b0a157e1b56b6566a34eb2798205cba35a211df3296ab7a74 \
+    --hash=sha256:f5b213bc29cc30a89a3130393b0e39c847a15d769d6e59539cd86b75d276b1a7
+    # via tensorflow-privacy
+scipy==1.9.3 \
+    --hash=sha256:06d2e1b4c491dc7d8eacea139a1b0b295f74e1a1a0f704c375028f8320d16e31 \
+    --hash=sha256:0d54222d7a3ba6022fdf5773931b5d7c56efe41ede7f7128c7b1637700409108 \
+    --hash=sha256:1884b66a54887e21addf9c16fb588720a8309a57b2e258ae1c7986d4444d3bc0 \
+    --hash=sha256:1a72d885fa44247f92743fc20732ae55564ff2a519e8302fb7e18717c5355a8b \
+    --hash=sha256:2318bef588acc7a574f5bfdff9c172d0b1bf2c8143d9582e05f878e580a3781e \
+    --hash=sha256:4db5b30849606a95dcf519763dd3ab6fe9bd91df49eba517359e450a7d80ce2e \
+    --hash=sha256:545c83ffb518094d8c9d83cce216c0c32f8c04aaf28b92cc8283eda0685162d5 \
+    --hash=sha256:5a04cd7d0d3eff6ea4719371cbc44df31411862b9646db617c99718ff68d4840 \
+    --hash=sha256:5b88e6d91ad9d59478fafe92a7c757d00c59e3bdc3331be8ada76a4f8d683f58 \
+    --hash=sha256:68239b6aa6f9c593da8be1509a05cb7f9efe98b80f43a5861cd24c7557e98523 \
+    --hash=sha256:83b89e9586c62e787f5012e8475fbb12185bafb996a03257e9675cd73d3736dd \
+    --hash=sha256:83c06e62a390a9167da60bedd4575a14c1f58ca9dfde59830fc42e5197283dab \
+    --hash=sha256:90453d2b93ea82a9f434e4e1cba043e779ff67b92f7a0e85d05d286a3625df3c \
+    --hash=sha256:abaf921531b5aeaafced90157db505e10345e45038c39e5d9b6c7922d68085cb \
+    --hash=sha256:b41bc822679ad1c9a5f023bc93f6d0543129ca0f37c1ce294dd9d386f0a21096 \
+    --hash=sha256:c68db6b290cbd4049012990d7fe71a2abd9ffbe82c0056ebe0f01df8be5436b0 \
+    --hash=sha256:cff3a5295234037e39500d35316a4c5794739433528310e117b8a9a0c76d20fc \
+    --hash=sha256:d01e1dd7b15bd2449c8bfc6b7cc67d630700ed655654f0dfcf121600bad205c9 \
+    --hash=sha256:d644a64e174c16cb4b2e41dfea6af722053e83d066da7343f333a54dae9bc31c \
+    --hash=sha256:da8245491d73ed0a994ed9c2e380fd058ce2fa8a18da204681f2fe1f57f98f95 \
+    --hash=sha256:fbc5c05c85c1a02be77b1ff591087c83bc44579c6d2bd9fb798bb64ea5e1a027
+    # via
+    #   -r requirements.in
+    #   dp-accounting
+    #   jax
+    #   jaxlib
+    #   scikit-learn
+    #   tensorflow-privacy
+six==1.16.0 \
+    --hash=sha256:1e61c37477a1626458e36f7b1d82aa5c9b094fa4802892072e49de9c60c4c926 \
+    --hash=sha256:8abb2f1d86890a2dfb989f9a77cfcfd3e47c2a354b01111771326f8aa26e0254
+    # via
+    #   astunparse
+    #   google-pasta
+    #   tensorboard
+    #   tensorflow
+    #   tensorflow-model-optimization
+    #   tensorflow-probability
+sqlalchemy==1.4.20 \
+    --hash=sha256:0f6d467b67a7e5048f1408e8ea60d6caa70be5b386d0eebbf1185ab49cb8c7e4 \
+    --hash=sha256:238d78b3110b7f7cffdb70bf9cda686e0d876a849bc78ba4d471aa7b1461f306 \
+    --hash=sha256:25c0e0f3a7e8c19350086b3c0fe93c4def045cec053d749ef15da710c4d54c81 \
+    --hash=sha256:2f60a2e599cf5cf5e5327ce60f2918b897e42ad9f405d10dd01e37869c0ce6fc \
+    --hash=sha256:38ee3a266afef2978e82824650457f70c5d74ec0cadec1b10fe5ed6f038eb5d0 \
+    --hash=sha256:46361690f1e1c5385994a4caeb6e8126063ff593a5c635700bbc1245de793c1e \
+    --hash=sha256:46b99eab618cdc1c871ea707b7c52edc23cfea6c750740cd242ba62b5c84de7f \
+    --hash=sha256:4a67371752fd86d1d03a3b82d4e75404608f6f4d579b9676124079a22a40c79f \
+    --hash=sha256:525dd3c2205b11a2bc6d770bf1ec63bde0253fd754b4c19c399d27ddc9dad0d3 \
+    --hash=sha256:6c8406c3d8c1c7d15da454de15d77f7bb48d14ede5db994f74226c348cf1050e \
+    --hash=sha256:6da83225a23eaf7b3f48f3d5f53c91b2cf00fbfa48b24a7a758160112dd3e123 \
+    --hash=sha256:7150e5b543b466f45f668b352f7abda27998cc8035f051d1b7e9524ca9eb2f5f \
+    --hash=sha256:76fbc24311a3d039d6cd147d396719f606d96d1413f3816c028a48e29367f646 \
+    --hash=sha256:854a7b15750e617e16f8d65dbc004f065a7963544b253b923f16109557648777 \
+    --hash=sha256:86c079732328f1add097b0b8079cd532b5d28e207fac93e9d6ea5f487506deef \
+    --hash=sha256:8d860c62e3f51623ccd528d8fac44580501df557d4b467cc5581587fcf057719 \
+    --hash=sha256:9675d5bc7e4f96a7bb2b54d14e9b269a5fb6e5d36ecc7d01f0f65bb9af3185f9 \
+    --hash=sha256:9841762d114018c49483c089fa2d47f7e612e57666323f615913d7d7f46e9606 \
+    --hash=sha256:9eb25bcf9161e2fcbe9eebe8e829719b2334e849183f0e496bf4b83722bcccfa \
+    --hash=sha256:aad3234a41340e9cf6184e621694e2a7233ba3f8aef9b1e6de8cba431b45ebd2 \
+    --hash=sha256:b502b5e2f08500cc4b8d29bfc4f51d805adcbc00f8d149e98fda8aae85ddb644 \
+    --hash=sha256:b86d83fefc8a8c394f3490c37e1953bc16c311a3d1d1cf91518793bfb9847fb4 \
+    --hash=sha256:c0eb2cd3ad4967fcbdd9e066e8cd91fe2c23c671dbae9952f0b4d3d42832cc5f \
+    --hash=sha256:e0d48456e1aa4f0537f9c9af7be71e1f0659ff68bc1cd538ebc785f6b007bd0d \
+    --hash=sha256:eaee5dd378f6f0d7c3ec49aeeb26564d55ac0ad73b9b4688bf29e66deabddf73 \
+    --hash=sha256:f14acb0fd16d404fda9370f93aace682f284340c89c3442ac747c5466ac7e2b5 \
+    --hash=sha256:f6fc526bd70898489d02bf52c8f0632ab377592ae954d0c0a5bb38d618dddaa9 \
+    --hash=sha256:fcd84e4d46a86291495d131a7824ba38d2e8278bda9425c50661a04633174319 \
+    --hash=sha256:ff38ecf89c69a531a7326c2dae71982edfe2f805f3c016cdc5bfd1a04ebf80cb \
+    --hash=sha256:ff8bebc7a9d297dff2003460e01db2c20c63818b45fb19170f388b1a72fe5a14
+    # via google-vizier
+tensorboard==2.17.0 \
+    --hash=sha256:859a499a9b1fb68a058858964486627100b71fcb21646861c61d31846a6478fb
+    # via tensorflow
+tensorboard-data-server==0.7.2 \
+    --hash=sha256:7e0610d205889588983836ec05dc098e80f97b7e7bbff7e994ebb78f578d0ddb \
+    --hash=sha256:9fe5d24221b29625dbc7328b0436ca7fc1c23de4acf4d272f1180856e32f9f60 \
+    --hash=sha256:ef687163c24185ae9754ed5650eb5bc4d84ff257aabdc33f0cc6f74d8ba54530
+    # via tensorboard
+tensorflow==2.17.0 \
+    --hash=sha256:0ad7bfea6afb4ded3928ca5b24df9fda876cea4904c103a5163fcc0c3483e7a4 \
+    --hash=sha256:147c93ded4cb7e500a65d3c26d74744ff41660db7a8afe2b00d1d08bf329b4ec \
+    --hash=sha256:278bc80642d799adf08dc4e04f291aab603bba7457d50c1f9bc191ebbca83f43 \
+    --hash=sha256:4ae8e6746deb2ec807b902ba26d62fcffb6a6b53555a1a5906ec00416c5e4175 \
+    --hash=sha256:515fe5ae8a9bc50312575412b08515f3ca66514c155078e0707bdffbea75d783 \
+    --hash=sha256:72adfef0ee39dd641627906fd7b244fcf21bdd8a87216a998ed74d9c74653aff \
+    --hash=sha256:8339777b1b5ebd8ffadaa8196f786e65fbb081a371d8e87b52f24563392d8552 \
+    --hash=sha256:8f80d11ad3766570deb6ff47d2bed2d166f51399ca08205e38ef024345571d6f \
+    --hash=sha256:97f89e95d68b4b46e1072243b9f315c3b340e27cc07b1e1988e2ca97ad844305 \
+    --hash=sha256:b36683ac28af20abc3a548c72bf4537b00df1b1f3dd39d59df3873fefaf26f15 \
+    --hash=sha256:ca82f98ea38fa6c9e08ccc69eb6c2fab5b35b30a8999115b8b63b6f02fc69d9d \
+    --hash=sha256:dde37cff74ed22b8fa2eea944805b001ae38e96adc989666422bdea34f4e2d47 \
+    --hash=sha256:e46090587f69e33637d17d7c3d94a790cac7d4bc5ff5ecbf3e71fdc6982fe96e \
+    --hash=sha256:e8d26d6c24ccfb139db1306599257ca8f5cfe254ef2d023bfb667f374a17a64d \
+    --hash=sha256:ee18b4fcd627c5e872eabb25092af6c808b6ec77948662c88fc5c89a60eb0211 \
+    --hash=sha256:ef615c133cf4d592a073feda634ccbeb521a554be57de74f8c318d38febbeab5
+    # via
+    #   -r requirements.in
+    #   tensorflow-privacy
+tensorflow-estimator==2.15.0 \
+    --hash=sha256:aedf21eec7fb2dc91150fc91a1ce12bc44dbb72278a08b58e79ff87c9e28f153
+    # via tensorflow-privacy
+tensorflow-io-gcs-filesystem==0.37.1 \
+    --hash=sha256:0df00891669390078a003cedbdd3b8e645c718b111917535fa1d7725e95cdb95 \
+    --hash=sha256:249c12b830165841411ba71e08215d0e94277a49c551e6dd5d72aab54fe5491b \
+    --hash=sha256:257aab23470a0796978efc9c2bcf8b0bc80f22e6298612a4c0a50d3f4e88060c \
+    --hash=sha256:286389a203a5aee1a4fa2e53718c661091aa5fea797ff4fa6715ab8436b02e6c \
+    --hash=sha256:32c50ab4e29a23c1f91cd0f9ab8c381a0ab10f45ef5c5252e94965916041737c \
+    --hash=sha256:426de1173cb81fbd62becec2012fc00322a295326d90eb6c737fab636f182aed \
+    --hash=sha256:6e1f2796b57e799a8ca1b75bf47c2aaa437c968408cc1a402a9862929e104cda \
+    --hash=sha256:8943036bbf84e7a2be3705cb56f9c9df7c48c9e614bb941f0936c58e3ca89d6f \
+    --hash=sha256:8febbfcc67c61e542a5ac1a98c7c20a91a5e1afc2e14b1ef0cb7c28bc3b6aa70 \
+    --hash=sha256:9679b36e3a80921876f31685ab6f7270f3411a4cc51bc2847e80d0e4b5291e27 \
+    --hash=sha256:b02f9c5f94fd62773954a04f69b68c4d576d076fd0db4ca25d5479f0fbfcdbad \
+    --hash=sha256:ee5da49019670ed364f3e5fb86b46420841a6c3cb52a300553c63841671b3e6d \
+    --hash=sha256:ee7c8ee5fe2fd8cb6392669ef16e71841133041fee8a330eff519ad9b36e4556 \
+    --hash=sha256:fbb33f1745f218464a59cecd9a18e32ca927b0f4d77abd8f8671b645cc1a182f \
+    --hash=sha256:fe8dcc6d222258a080ac3dfcaaaa347325ce36a7a046277f6b3e19abc1efb3c5 \
+    --hash=sha256:ffebb6666a7bfc28005f4fbbb111a455b5e7d6cd3b12752b7050863ecb27d5cc
+    # via tensorflow
+tensorflow-model-optimization==0.7.5 \
+    --hash=sha256:cfd188188e860d2e3994185a423677888a5e042dfa4fb7f9611e9fa3cf990fc9
+    # via -r requirements.in
+tensorflow-privacy==0.9.0 \
+    --hash=sha256:2c035ee32400011905a8e6690626e16da83cc2899cba746dcdbd4c7ec2b3df10
+    # via -r requirements.in
+tensorflow-probability==0.22.1 \
+    --hash=sha256:3035b936b028ea10bd3a9589329557f5b2c5ace813a6bff3f59acfe3226eef9b
+    # via tensorflow-privacy
+termcolor==2.4.0 \
+    --hash=sha256:9297c0df9c99445c2412e832e882a7884038a25617c60cea2ad69488d4040d63 \
+    --hash=sha256:aab9e56047c8ac41ed798fa36d892a37aca6b3e9159f3e0c24bc64a9b3ac7b7a
+    # via tensorflow
+threadpoolctl==3.5.0 \
+    --hash=sha256:082433502dd922bf738de0d8bcc4fdcbf0979ff44c42bd40f5af8a282f6fa107 \
+    --hash=sha256:56c1e26c150397e58c4926da8eeee87533b1e32bef131bd4bf6a2f45f3185467
+    # via scikit-learn
+tqdm==4.66.4 \
+    --hash=sha256:b75ca56b413b030bc3f00af51fd2c1a1a5eac6a0c1cca83cbb37a5c52abce644 \
+    --hash=sha256:e4d936c9de8727928f3be6079590e97d9abfe8d39a590be678eb5919ffc186bb
+    # via -r requirements.in
+typing-extensions==4.5.0 \
+    --hash=sha256:5cb5f4a79139d699607b3ef622a1dedafa84e115ab0024e0d9c044a9479ca7cb \
+    --hash=sha256:fb33085c39dd998ac16d1431ebc293a8b3eedd00fd4a32de0ff79002c19511b4
+    # via
+    #   -r requirements.in
+    #   optree
+    #   tensorflow
+urllib3==2.2.2 \
+    --hash=sha256:a448b2f64d686155468037e1ace9f2d2199776e17f0a46610480d311f73e3472 \
+    --hash=sha256:dd505485549a7a552833da5e6063639d0d177c04f23bc3864e41e5dc5f612168
+    # via requests
+werkzeug==3.0.3 \
+    --hash=sha256:097e5bfda9f0aba8da6b8545146def481d06aa7d3266e7448e2cccf67dd8bd18 \
+    --hash=sha256:fc9645dc43e03e4d630d23143a04a7f947a9a3b5727cd535fdfe155a17cc48c8
+    # via tensorboard
+wheel==0.43.0 \
+    --hash=sha256:465ef92c69fa5c5da2d1cf8ac40559a8c940886afcef87dcf14b9470862f1d85 \
+    --hash=sha256:55c570405f142630c6b9f72fe09d9b67cf1477fcf543ae5b8dcb1f5b7377da81
+    # via astunparse
+wrapt==1.16.0 \
+    --hash=sha256:0d2691979e93d06a95a26257adb7bfd0c93818e89b1406f5a28f36e0d8c1e1fc \
+    --hash=sha256:14d7dc606219cdd7405133c713f2c218d4252f2a469003f8c46bb92d5d095d81 \
+    --hash=sha256:1a5db485fe2de4403f13fafdc231b0dbae5eca4359232d2efc79025527375b09 \
+    --hash=sha256:1acd723ee2a8826f3d53910255643e33673e1d11db84ce5880675954183ec47e \
+    --hash=sha256:1ca9b6085e4f866bd584fb135a041bfc32cab916e69f714a7d1d397f8c4891ca \
+    --hash=sha256:1dd50a2696ff89f57bd8847647a1c363b687d3d796dc30d4dd4a9d1689a706f0 \
+    --hash=sha256:2076fad65c6736184e77d7d4729b63a6d1ae0b70da4868adeec40989858eb3fb \
+    --hash=sha256:2a88e6010048489cda82b1326889ec075a8c856c2e6a256072b28eaee3ccf487 \
+    --hash=sha256:3ebf019be5c09d400cf7b024aa52b1f3aeebeff51550d007e92c3c1c4afc2a40 \
+    --hash=sha256:418abb18146475c310d7a6dc71143d6f7adec5b004ac9ce08dc7a34e2babdc5c \
+    --hash=sha256:43aa59eadec7890d9958748db829df269f0368521ba6dc68cc172d5d03ed8060 \
+    --hash=sha256:44a2754372e32ab315734c6c73b24351d06e77ffff6ae27d2ecf14cf3d229202 \
+    --hash=sha256:490b0ee15c1a55be9c1bd8609b8cecd60e325f0575fc98f50058eae366e01f41 \
+    --hash=sha256:49aac49dc4782cb04f58986e81ea0b4768e4ff197b57324dcbd7699c5dfb40b9 \
+    --hash=sha256:5eb404d89131ec9b4f748fa5cfb5346802e5ee8836f57d516576e61f304f3b7b \
+    --hash=sha256:5f15814a33e42b04e3de432e573aa557f9f0f56458745c2074952f564c50e664 \
+    --hash=sha256:5f370f952971e7d17c7d1ead40e49f32345a7f7a5373571ef44d800d06b1899d \
+    --hash=sha256:66027d667efe95cc4fa945af59f92c5a02c6f5bb6012bff9e60542c74c75c362 \
+    --hash=sha256:66dfbaa7cfa3eb707bbfcd46dab2bc6207b005cbc9caa2199bcbc81d95071a00 \
+    --hash=sha256:685f568fa5e627e93f3b52fda002c7ed2fa1800b50ce51f6ed1d572d8ab3e7fc \
+    --hash=sha256:6906c4100a8fcbf2fa735f6059214bb13b97f75b1a61777fcf6432121ef12ef1 \
+    --hash=sha256:6a42cd0cfa8ffc1915aef79cb4284f6383d8a3e9dcca70c445dcfdd639d51267 \
+    --hash=sha256:6dcfcffe73710be01d90cae08c3e548d90932d37b39ef83969ae135d36ef3956 \
+    --hash=sha256:6f6eac2360f2d543cc875a0e5efd413b6cbd483cb3ad7ebf888884a6e0d2e966 \
+    --hash=sha256:72554a23c78a8e7aa02abbd699d129eead8b147a23c56e08d08dfc29cfdddca1 \
+    --hash=sha256:73870c364c11f03ed072dda68ff7aea6d2a3a5c3fe250d917a429c7432e15228 \
+    --hash=sha256:73aa7d98215d39b8455f103de64391cb79dfcad601701a3aa0dddacf74911d72 \
+    --hash=sha256:75ea7d0ee2a15733684badb16de6794894ed9c55aa5e9903260922f0482e687d \
+    --hash=sha256:7bd2d7ff69a2cac767fbf7a2b206add2e9a210e57947dd7ce03e25d03d2de292 \
+    --hash=sha256:807cc8543a477ab7422f1120a217054f958a66ef7314f76dd9e77d3f02cdccd0 \
+    --hash=sha256:8e9723528b9f787dc59168369e42ae1c3b0d3fadb2f1a71de14531d321ee05b0 \
+    --hash=sha256:9090c9e676d5236a6948330e83cb89969f433b1943a558968f659ead07cb3b36 \
+    --hash=sha256:9153ed35fc5e4fa3b2fe97bddaa7cbec0ed22412b85bcdaf54aeba92ea37428c \
+    --hash=sha256:9159485323798c8dc530a224bd3ffcf76659319ccc7bbd52e01e73bd0241a0c5 \
+    --hash=sha256:941988b89b4fd6b41c3f0bfb20e92bd23746579736b7343283297c4c8cbae68f \
+    --hash=sha256:94265b00870aa407bd0cbcfd536f17ecde43b94fb8d228560a1e9d3041462d73 \
+    --hash=sha256:98b5e1f498a8ca1858a1cdbffb023bfd954da4e3fa2c0cb5853d40014557248b \
+    --hash=sha256:9b201ae332c3637a42f02d1045e1d0cccfdc41f1f2f801dafbaa7e9b4797bfc2 \
+    --hash=sha256:a0ea261ce52b5952bf669684a251a66df239ec6d441ccb59ec7afa882265d593 \
+    --hash=sha256:a33a747400b94b6d6b8a165e4480264a64a78c8a4c734b62136062e9a248dd39 \
+    --hash=sha256:a452f9ca3e3267cd4d0fcf2edd0d035b1934ac2bd7e0e57ac91ad6b95c0c6389 \
+    --hash=sha256:a86373cf37cd7764f2201b76496aba58a52e76dedfaa698ef9e9688bfd9e41cf \
+    --hash=sha256:ac83a914ebaf589b69f7d0a1277602ff494e21f4c2f743313414378f8f50a4cf \
+    --hash=sha256:aefbc4cb0a54f91af643660a0a150ce2c090d3652cf4052a5397fb2de549cd89 \
+    --hash=sha256:b3646eefa23daeba62643a58aac816945cadc0afaf21800a1421eeba5f6cfb9c \
+    --hash=sha256:b47cfad9e9bbbed2339081f4e346c93ecd7ab504299403320bf85f7f85c7d46c \
+    --hash=sha256:b935ae30c6e7400022b50f8d359c03ed233d45b725cfdd299462f41ee5ffba6f \
+    --hash=sha256:bb2dee3874a500de01c93d5c71415fcaef1d858370d405824783e7a8ef5db440 \
+    --hash=sha256:bc57efac2da352a51cc4658878a68d2b1b67dbe9d33c36cb826ca449d80a8465 \
+    --hash=sha256:bf5703fdeb350e36885f2875d853ce13172ae281c56e509f4e6eca049bdfb136 \
+    --hash=sha256:c31f72b1b6624c9d863fc095da460802f43a7c6868c5dda140f51da24fd47d7b \
+    --hash=sha256:c5cd603b575ebceca7da5a3a251e69561bec509e0b46e4993e1cac402b7247b8 \
+    --hash=sha256:d2efee35b4b0a347e0d99d28e884dfd82797852d62fcd7ebdeee26f3ceb72cf3 \
+    --hash=sha256:d462f28826f4657968ae51d2181a074dfe03c200d6131690b7d65d55b0f360f8 \
+    --hash=sha256:d5e49454f19ef621089e204f862388d29e6e8d8b162efce05208913dde5b9ad6 \
+    --hash=sha256:da4813f751142436b075ed7aa012a8778aa43a99f7b36afe9b742d3ed8bdc95e \
+    --hash=sha256:db2e408d983b0e61e238cf579c09ef7020560441906ca990fe8412153e3b291f \
+    --hash=sha256:db98ad84a55eb09b3c32a96c576476777e87c520a34e2519d3e59c44710c002c \
+    --hash=sha256:dbed418ba5c3dce92619656802cc5355cb679e58d0d89b50f116e4a9d5a9603e \
+    --hash=sha256:dcdba5c86e368442528f7060039eda390cc4091bfd1dca41e8046af7c910dda8 \
+    --hash=sha256:decbfa2f618fa8ed81c95ee18a387ff973143c656ef800c9f24fb7e9c16054e2 \
+    --hash=sha256:e4fdb9275308292e880dcbeb12546df7f3e0f96c6b41197e0cf37d2826359020 \
+    --hash=sha256:eb1b046be06b0fce7249f1d025cd359b4b80fc1c3e24ad9eca33e0dcdb2e4a35 \
+    --hash=sha256:eb6e651000a19c96f452c85132811d25e9264d836951022d6e81df2fff38337d \
+    --hash=sha256:ed867c42c268f876097248e05b6117a65bcd1e63b779e916fe2e33cd6fd0d3c3 \
+    --hash=sha256:edfad1d29c73f9b863ebe7082ae9321374ccb10879eeabc84ba3b69f2579d537 \
+    --hash=sha256:f2058f813d4f2b5e3a9eb2eb3faf8f1d99b81c3e51aeda4b168406443e8ba809 \
+    --hash=sha256:f6b2d0c6703c988d334f297aa5df18c45e97b0af3679bb75059e0e0bd8b1069d \
+    --hash=sha256:f8212564d49c50eb4565e502814f694e240c55551a5f1bc841d4fcaabb0a9b8a \
+    --hash=sha256:ffa565331890b90056c01db69c0fe634a776f8019c143a5ae265f9c6bc4bd6d4
+    # via tensorflow
+
+# The following packages are considered to be unsafe in a requirements file:
+setuptools==72.1.0 \
+    --hash=sha256:5a03e1860cf56bb6ef48ce186b0e557fdba433237481a9a625176c2831be15d1 \
+    --hash=sha256:8d243eff56d095e5817f796ede6ae32941278f542e0f941867cc05ae52b162ec
+    # via
+    #   grpcio-tools
+    #   tensorboard
+    #   tensorflow
diff --git a/tensorflow_federated/cc/core/impl/aggregation/base/BUILD b/tensorflow_federated/cc/core/impl/aggregation/base/BUILD
index 9aa783da01..3f92ce24de 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/base/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/base/BUILD
@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+load("//tools:build_defs.bzl", "tff_cc_library", "tff_cc_test")
+
 package(
     default_applicable_licenses = ["//:package_license"],
     default_visibility = ["//tensorflow_federated/cc/core/impl/aggregation:__subpackages__"],
 )
 
-cc_library(
+tff_cc_library(
     name = "base",
     srcs = [
         "base_name.cc",
@@ -41,7 +43,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "monitoring_test",
     size = "small",
     srcs = ["monitoring_test.cc"],
@@ -51,7 +53,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "platform_test",
     size = "small",
     srcs = ["platform_test.cc"],
@@ -63,7 +65,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "scheduler",
     srcs = ["scheduler.cc"],
     hdrs = ["scheduler.h"],
@@ -74,7 +76,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "scheduler_test",
     size = "small",
     srcs = ["scheduler_test.cc"],
@@ -86,7 +88,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "move_to_lambda_test",
     size = "small",
     srcs = ["move_to_lambda_test.cc"],
@@ -97,12 +99,12 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "unique_value",
     hdrs = ["unique_value.h"],
 )
 
-cc_test(
+tff_cc_test(
     name = "unique_value_test",
     srcs = ["unique_value_test.cc"],
     deps = [
@@ -112,7 +114,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "clock",
     srcs = ["clock.cc"],
     hdrs = ["clock.h"],
@@ -125,7 +127,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "simulated_clock",
     testonly = 1,
     srcs = ["simulated_clock.cc"],
@@ -138,7 +140,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "simulated_clock_test",
     srcs = ["simulated_clock_test.cc"],
     deps = [
diff --git a/tensorflow_federated/cc/core/impl/aggregation/core/BUILD b/tensorflow_federated/cc/core/impl/aggregation/core/BUILD
index 704001b424..d0f18893a4 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/core/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/core/BUILD
@@ -14,6 +14,7 @@
 
 load("@com_github_grpc_grpc//bazel:python_rules.bzl", "py_proto_library")
 load("@rules_cc//cc:defs.bzl", "cc_proto_library")
+load("//tools:build_defs.bzl", "tff_cc_binary", "tff_cc_library", "tff_cc_test")
 
 package(
     default_applicable_licenses = ["//:package_license"],
@@ -55,7 +56,7 @@ cc_proto_library(
     deps = [":agg_core_proto"],
 )
 
-cc_library(
+tff_cc_library(
     name = "tensor",
     srcs = [
         "datatype.cc",
@@ -86,7 +87,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "aggregator",
     srcs = [
         "tensor_aggregator.cc",
@@ -112,14 +113,14 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "intrinsic",
     hdrs = ["intrinsic.h"],
     visibility = ["//visibility:public"],
     deps = [":tensor"],
 )
 
-cc_library(
+tff_cc_library(
     name = "dp_noise_mechanisms",
     srcs = ["dp_noise_mechanisms.cc"],
     hdrs = [
@@ -134,7 +135,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dp_noise_mechanisms_test",
     srcs = ["dp_noise_mechanisms_test.cc"],
     deps = [
@@ -147,7 +148,7 @@ cc_test(
 )
 
 # TODO: b/352020454 - Create one library per cc & hh pair. Make them aggregation_cores deps.
-cc_library(
+tff_cc_library(
     name = "aggregation_cores",
     srcs = [
         "composite_key_combiner.cc",
@@ -194,7 +195,7 @@ cc_library(
     alwayslink = 1,
 )
 
-cc_test(
+tff_cc_test(
     name = "tensor_test",
     srcs = [
         "tensor_data_test.cc",
@@ -214,7 +215,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "agg_vector_test",
     srcs = ["agg_vector_test.cc"],
     deps = [
@@ -224,7 +225,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "aggregator_test",
     srcs = ["agg_vector_aggregator_test.cc"],
     deps = [
@@ -240,7 +241,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "tensor_aggregator_registry_test",
     srcs = ["tensor_aggregator_registry_test.cc"],
     deps = [
@@ -252,7 +253,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "mutable_string_data",
     hdrs = ["mutable_string_data.h"],
     visibility = ["//visibility:public"],
@@ -262,17 +263,17 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "fedsql_constants",
     hdrs = ["fedsql_constants.h"],
 )
 
-cc_library(
+tff_cc_library(
     name = "dp_fedsql_constants",
     hdrs = ["dp_fedsql_constants.h"],
 )
 
-cc_test(
+tff_cc_test(
     name = "federated_mean_test",
     srcs = ["federated_mean_test.cc"],
     deps = [
@@ -289,7 +290,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "federated_sum_test",
     srcs = ["federated_sum_test.cc"],
     deps = [
@@ -306,7 +307,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "input_tensor_list_test",
     srcs = ["input_tensor_list_test.cc"],
     deps = [
@@ -316,7 +317,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "composite_key_combiner_test",
     srcs = ["composite_key_combiner_test.cc"],
     deps = [
@@ -332,7 +333,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dp_composite_key_combiner_test",
     srcs = ["dp_composite_key_combiner_test.cc"],
     deps = [
@@ -349,7 +350,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "mutable_vector_data_test",
     srcs = ["mutable_vector_data_test.cc"],
     deps = [
@@ -359,7 +360,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "vector_string_data_test",
     srcs = [
         "vector_string_data.h",
@@ -372,7 +373,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "mutable_string_data_test",
     srcs = ["mutable_string_data_test.cc"],
     deps = [
@@ -384,7 +385,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "one_dim_grouping_aggregator_test",
     srcs = ["one_dim_grouping_aggregator_test.cc"],
     deps = [
@@ -399,7 +400,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "group_by_aggregator_test",
     srcs = ["group_by_aggregator_test.cc"],
     deps = [
@@ -416,7 +417,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dp_open_domain_histogram_test",
     srcs = ["dp_open_domain_histogram_test.cc"],
     deps = [
@@ -434,7 +435,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dp_closed_domain_histogram_test",
     srcs = ["dp_closed_domain_histogram_test.cc"],
     deps = [
@@ -453,7 +454,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "grouping_federated_sum_test",
     srcs = ["grouping_federated_sum_test.cc"],
     deps = [
@@ -469,7 +470,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dp_grouping_federated_sum_test",
     srcs = ["dp_grouping_federated_sum_test.cc"],
     deps = [
@@ -488,7 +489,7 @@ cc_test(
     ],
 )
 
-cc_binary(
+tff_cc_binary(
     name = "federated_sum_bench",
     testonly = 1,
     srcs = ["federated_sum_bench.cc"],
@@ -502,7 +503,7 @@ cc_binary(
     ],
 )
 
-cc_binary(
+tff_cc_binary(
     name = "group_by_bench",
     testonly = 1,
     srcs = ["group_by_bench.cc"],
diff --git a/tensorflow_federated/cc/core/impl/aggregation/protocol/BUILD b/tensorflow_federated/cc/core/impl/aggregation/protocol/BUILD
index cbe8cada5a..d18c603879 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/protocol/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/protocol/BUILD
@@ -14,6 +14,7 @@
 
 load("@com_github_grpc_grpc//bazel:python_rules.bzl", "py_proto_library")
 load("@rules_cc//cc:defs.bzl", "cc_proto_library")
+load("//tools:build_defs.bzl", "tff_cc_library", "tff_cc_test")
 
 package(
     default_applicable_licenses = ["//:package_license"],
@@ -62,7 +63,7 @@ py_proto_library(
     deps = [":configuration_proto"],
 )
 
-cc_library(
+tff_cc_library(
     name = "aggregation_protocol",
     hdrs = ["aggregation_protocol.h"],
     deps = [
@@ -73,7 +74,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "resource_resolver",
     hdrs = ["resource_resolver.h"],
     deps = [
@@ -82,7 +83,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "checkpoint_builder",
     hdrs = ["checkpoint_builder.h"],
     deps = [
@@ -93,7 +94,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "checkpoint_parser",
     hdrs = ["checkpoint_parser.h"],
     deps = [
@@ -103,7 +104,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "checkpoint_aggregator",
     srcs = ["checkpoint_aggregator.cc"],
     hdrs = ["checkpoint_aggregator.h"],
@@ -129,7 +130,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "checkpoint_aggregator_test",
     srcs = ["checkpoint_aggregator_test.cc"],
     deps = [
@@ -153,7 +154,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "config_converter",
     srcs = ["config_converter.cc"],
     hdrs = ["config_converter.h"],
@@ -172,7 +173,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "federated_compute_checkpoint_builder",
     srcs = ["federated_compute_checkpoint_builder.cc"],
     hdrs = ["federated_compute_checkpoint_builder.h"],
@@ -188,7 +189,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "federated_compute_checkpoint_builder_test",
     srcs = ["federated_compute_checkpoint_builder_test.cc"],
     deps = [
@@ -206,7 +207,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "federated_compute_checkpoint_parser",
     srcs = ["federated_compute_checkpoint_parser.cc"],
     hdrs = ["federated_compute_checkpoint_parser.h"],
@@ -226,12 +227,12 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "checkpoint_header",
     hdrs = ["checkpoint_header.h"],
 )
 
-cc_test(
+tff_cc_test(
     name = "federated_compute_checkpoint_parser_test",
     srcs = ["federated_compute_checkpoint_parser_test.cc"],
     deps = [
@@ -249,7 +250,7 @@ cc_test(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "config_converter_test",
     srcs = ["config_converter_test.cc"],
     deps = [
diff --git a/tensorflow_federated/cc/core/impl/aggregation/protocol/simple_aggregation/BUILD b/tensorflow_federated/cc/core/impl/aggregation/protocol/simple_aggregation/BUILD
index a3b1388cd3..0efc0833dc 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/protocol/simple_aggregation/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/protocol/simple_aggregation/BUILD
@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+load("//tools:build_defs.bzl", "tff_cc_library", "tff_cc_test")
+
 package(
     default_applicable_licenses = ["//:package_license"],
     default_visibility = ["//visibility:public"],
 )
 
-cc_library(
+tff_cc_library(
     name = "latency_aggregator",
     hdrs = ["latency_aggregator.h"],
     deps = [
@@ -27,7 +29,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "latency_aggregator_test",
     srcs = ["latency_aggregator_test.cc"],
     deps = [
@@ -39,7 +41,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "cancelable_callback",
     srcs = ["cancelable_callback.cc"],
     hdrs = ["cancelable_callback.h"],
@@ -51,7 +53,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "cancelable_callback_test",
     srcs = ["cancelable_callback_test.cc"],
     deps = [
@@ -64,7 +66,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "simple_aggregation",
     srcs = ["simple_aggregation_protocol.cc"],
     hdrs = ["simple_aggregation_protocol.h"],
@@ -96,7 +98,7 @@ cc_library(
     alwayslink = 1,
 )
 
-cc_test(
+tff_cc_test(
     name = "simple_aggregation_test",
     srcs = ["simple_aggregation_protocol_test.cc"],
     deps = [
diff --git a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/BUILD b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/BUILD
index d5b3eb12b4..244f8195e1 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/BUILD
@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+load("//tools:build_defs.bzl", "tff_cc_library", "tff_cc_test")
+
 package(
     default_applicable_licenses = ["//:package_license"],
     default_visibility = ["//visibility:public"],
 )
 
-cc_library(
+tff_cc_library(
     name = "converters",
     srcs = ["converters.cc"],
     hdrs = ["converters.h"],
@@ -29,14 +31,14 @@ cc_library(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
+        "@local_tsl//tsl/platform:refcount",
         "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core:protos_all_cc",
         "@org_tensorflow//tensorflow/core/platform:tstring",
-        "@org_tensorflow//tensorflow/tsl/platform:refcount",
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "converters_test",
     srcs = ["converters_test.cc"],
     deps = [
@@ -56,7 +58,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "checkpoint_reader",
     srcs = ["checkpoint_reader.cc"],
     hdrs = ["checkpoint_reader.h"],
@@ -76,7 +78,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "checkpoint_reader_test",
     srcs = ["checkpoint_reader_test.cc"],
     deps = [
@@ -90,7 +92,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "checkpoint_writer",
     srcs = ["checkpoint_writer.cc"],
     hdrs = ["checkpoint_writer.h"],
@@ -103,7 +105,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "checkpoint_writer_test",
     srcs = ["checkpoint_writer_test.cc"],
     deps = [
@@ -116,7 +118,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "tensorflow_checkpoint_builder_factory",
     srcs = ["tensorflow_checkpoint_builder_factory.cc"],
     hdrs = ["tensorflow_checkpoint_builder_factory.h"],
@@ -130,12 +132,12 @@ cc_library(
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
         "@com_google_absl//absl/strings:cord",
+        "@local_tsl//tsl/platform:env",
         "@org_tensorflow//tensorflow/core/platform:env",
-        "@org_tensorflow//tensorflow/tsl/platform:env",
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "tensorflow_checkpoint_builder_factory_test",
     srcs = ["tensorflow_checkpoint_builder_factory_test.cc"],
     deps = [
@@ -151,7 +153,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "tensorflow_checkpoint_parser_factory",
     srcs = ["tensorflow_checkpoint_parser_factory.cc"],
     hdrs = ["tensorflow_checkpoint_parser_factory.h"],
@@ -165,12 +167,12 @@ cc_library(
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
         "@com_google_absl//absl/strings:cord",
+        "@local_tsl//tsl/platform:env",
         "@org_tensorflow//tensorflow/core/platform:env",
-        "@org_tensorflow//tensorflow/tsl/platform:env",
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "tensorflow_checkpoint_parser_factory_test",
     srcs = ["tensorflow_checkpoint_parser_factory_test.cc"],
     deps = [
diff --git a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/converters.cc b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/converters.cc
index 3db0cab4f2..098bae0c5c 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/converters.cc
+++ b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/converters.cc
@@ -29,7 +29,7 @@
 #include "tensorflow/core/framework/tensor_shape.h"
 #include "tensorflow/core/framework/tensor_shape.pb.h"
 #include "tensorflow/core/platform/tstring.h"
-#include "tensorflow/tsl/platform/refcount.h"
+#include "tsl/platform/refcount.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/base/monitoring.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/datatype.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/tensor.h"
diff --git a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/python/BUILD b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/python/BUILD
index d8c21e4729..03b44649f9 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/python/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/python/BUILD
@@ -53,5 +53,7 @@ py_test(
         "//tensorflow_federated/cc/core/impl/aggregation/core:tensor_py_pb2",
         "//tensorflow_federated/cc/core/impl/aggregation/protocol:aggregation_protocol_messages_py_pb2",
         "//tensorflow_federated/cc/core/impl/aggregation/protocol:configuration_py_pb2",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_builder_factory.cc b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_builder_factory.cc
index 3903a74d36..1febef4025 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_builder_factory.cc
+++ b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_builder_factory.cc
@@ -29,7 +29,7 @@
 #include "absl/strings/str_cat.h"
 #include "absl/strings/string_view.h"
 #include "tensorflow/core/platform/file_system.h"
-#include "tensorflow/tsl/platform/env.h"
+#include "tsl/platform/env.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/base/monitoring.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/tensor.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/protocol/checkpoint_builder.h"
diff --git a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_parser_factory.cc b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_parser_factory.cc
index dd21957730..ce736d63bb 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_parser_factory.cc
+++ b/tensorflow_federated/cc/core/impl/aggregation/tensorflow/tensorflow_checkpoint_parser_factory.cc
@@ -29,7 +29,7 @@
 #include "absl/strings/str_cat.h"
 #include "absl/strings/string_view.h"
 #include "tensorflow/core/platform/file_system.h"
-#include "tensorflow/tsl/platform/env.h"
+#include "tsl/platform/env.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/base/monitoring.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/core/tensor.h"
 #include "tensorflow_federated/cc/core/impl/aggregation/protocol/checkpoint_parser.h"
diff --git a/tensorflow_federated/cc/core/impl/aggregation/testing/BUILD b/tensorflow_federated/cc/core/impl/aggregation/testing/BUILD
index 7dc6a85d74..b8d8614188 100644
--- a/tensorflow_federated/cc/core/impl/aggregation/testing/BUILD
+++ b/tensorflow_federated/cc/core/impl/aggregation/testing/BUILD
@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+load("//tools:build_defs.bzl", "tff_cc_library")
+
 package(
     default_applicable_licenses = ["//:package_license"],
     default_visibility = ["//visibility:public"],
 )
 
-cc_library(
+tff_cc_library(
     name = "testing",
     testonly = True,
     srcs = ["testing.cc"],
@@ -49,7 +51,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "mocks",
     testonly = True,
     hdrs = ["mocks.h"],
@@ -65,7 +67,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "parse_text_proto",
     testonly = True,
     hdrs = ["parse_text_proto.h"],
@@ -75,7 +77,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "test_data",
     testonly = True,
     hdrs = ["test_data.h"],
diff --git a/tensorflow_federated/cc/core/impl/executor_stacks/BUILD b/tensorflow_federated/cc/core/impl/executor_stacks/BUILD
index f943042cb7..d0b002ef51 100644
--- a/tensorflow_federated/cc/core/impl/executor_stacks/BUILD
+++ b/tensorflow_federated/cc/core/impl/executor_stacks/BUILD
@@ -1,4 +1,5 @@
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+load("//tools:build_defs.bzl", "tff_cc_library", "tff_cc_test")
 
 package(
     default_applicable_licenses = ["//:package_license"],
@@ -33,7 +34,7 @@ pybind_extension(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "local_stacks",
     srcs = ["local_stacks.cc"],
     hdrs = ["local_stacks.h"],
@@ -49,7 +50,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "local_stacks_test",
     srcs = ["local_stacks_test.cc"],
     deps = [
@@ -63,7 +64,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "remote_stacks",
     srcs = ["remote_stacks.cc"],
     hdrs = ["remote_stacks.h"],
@@ -86,7 +87,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "remote_stacks_test",
     srcs = ["remote_stacks_test.cc"],
     deps = [
diff --git a/tensorflow_federated/cc/core/impl/executors/BUILD b/tensorflow_federated/cc/core/impl/executors/BUILD
index b87b798e4b..adcfff8f15 100644
--- a/tensorflow_federated/cc/core/impl/executors/BUILD
+++ b/tensorflow_federated/cc/core/impl/executors/BUILD
@@ -1,6 +1,6 @@
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
 load("@rules_python//python:defs.bzl", "py_binary")
-load("//tools:build_defs.bzl", "tff_cc_cpu_gpu_test")
+load("//tools:build_defs.bzl", "tff_cc_cpu_gpu_test", "tff_cc_library", "tff_cc_test")
 
 package(
     default_applicable_licenses = ["//:package_license"],
@@ -21,14 +21,14 @@ package_group(
 
 licenses(["notice"])
 
-cc_library(
+tff_cc_library(
     name = "array_shape_test_utils",
     testonly = True,
     hdrs = ["array_shape_test_utils.h"],
     deps = ["//tensorflow_federated/proto/v0:array_cc_proto"],
 )
 
-cc_library(
+tff_cc_library(
     name = "array_test_utils",
     testonly = True,
     hdrs = ["array_test_utils.h"],
@@ -42,7 +42,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "cardinalities",
     srcs = ["cardinalities.cc"],
     hdrs = ["cardinalities.h"],
@@ -54,7 +54,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "composing_executor",
     srcs = ["composing_executor.cc"],
     hdrs = ["composing_executor.h"],
@@ -81,7 +81,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "composing_executor_test",
     srcs = ["composing_executor_test.cc"],
     deps = [
@@ -104,13 +104,13 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "computations",
     hdrs = ["computations.h"],
     deps = ["//tensorflow_federated/proto/v0:computation_cc_proto"],
 )
 
-cc_library(
+tff_cc_library(
     name = "data_backend",
     hdrs = ["data_backend.h"],
     visibility = ["//visibility:public"],
@@ -123,7 +123,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "data_executor",
     srcs = ["data_executor.cc"],
     hdrs = ["data_executor.h"],
@@ -137,7 +137,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "data_executor_test",
     srcs = ["data_executor_test.cc"],
     deps = [
@@ -155,7 +155,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "dataset_conversions",
     srcs = ["dataset_conversions.cc"],
     hdrs = ["dataset_conversions.h"],
@@ -168,7 +168,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dataset_conversions_test",
     srcs = ["dataset_conversions_test.cc"],
     deps = [
@@ -187,7 +187,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "dataset_from_tensor_structures",
     srcs = ["dataset_from_tensor_structures.cc"],
     hdrs = ["dataset_from_tensor_structures.h"],
@@ -208,7 +208,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "dataset_from_tensor_structures_test",
     srcs = ["dataset_from_tensor_structures_test.cc"],
     args = [
@@ -233,7 +233,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "eager_computation",
     srcs = ["eager_computation.cc"],
     hdrs = ["eager_computation.h"],
@@ -291,7 +291,7 @@ tff_cc_cpu_gpu_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "executor",
     srcs = ["executor.cc"],
     hdrs = ["executor.h"],
@@ -343,14 +343,13 @@ pybind_extension(
         "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core/platform:status",
         "@org_tensorflow//tensorflow/core/platform:strcat",
-        "@org_tensorflow//tensorflow/python/lib/core:ndarray_tensor_headers",
         "@pybind11_abseil//pybind11_abseil:absl_casters",
         "@pybind11_abseil//pybind11_abseil:status_casters",
         "@pybind11_protobuf//pybind11_protobuf:native_proto_caster",
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "executor_service",
     srcs = ["executor_service.cc"],
     hdrs = ["executor_service.h"],
@@ -373,7 +372,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "executor_service_test",
     srcs = ["executor_service_test.cc"],
     deps = [
@@ -395,7 +394,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "executor_test_base",
     testonly = True,
     srcs = [],
@@ -420,7 +419,7 @@ pybind_extension(
     deps = [":mock_executor"],
 )
 
-cc_library(
+tff_cc_library(
     name = "federated_intrinsics",
     srcs = ["federated_intrinsics.cc"],
     hdrs = ["federated_intrinsics.h"],
@@ -431,7 +430,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "federating_executor",
     srcs = ["federating_executor.cc"],
     hdrs = ["federating_executor.h"],
@@ -460,7 +459,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "federating_executor_test",
     srcs = ["federating_executor_test.cc"],
     deps = [
@@ -496,6 +495,10 @@ py_binary(
     name = "make_reduce_lambda_test_graph",
     testonly = True,
     srcs = ["make_reduce_lambda_test_graph.py"],
+    deps = [
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 genrule(
@@ -513,9 +516,13 @@ py_binary(
     name = "make_structural_reduce_test_graph",
     testonly = True,
     srcs = ["make_structural_reduce_test_graph.py"],
+    deps = [
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
-cc_library(
+tff_cc_library(
     name = "mock_data_backend",
     testonly = True,
     hdrs = ["mock_data_backend.h"],
@@ -529,7 +536,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "mock_executor",
     testonly = True,
     srcs = ["mock_executor.cc"],
@@ -546,7 +553,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "mock_grpc",
     testonly = True,
     hdrs = ["mock_grpc.h"],
@@ -554,13 +561,12 @@ cc_library(
         "//tensorflow_federated/proto/v0:executor_cc_grpc_proto",
         "//tensorflow_federated/proto/v0:executor_cc_proto",
         "@com_github_grpc_grpc//:grpc++",
-        "@com_github_grpc_grpc//:grpc_security_base",
         "@com_google_absl//absl/strings",
         "@com_google_googletest//:gtest",
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "reference_resolving_executor",
     srcs = ["reference_resolving_executor.cc"],
     hdrs = ["reference_resolving_executor.h"],
@@ -578,7 +584,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "reference_resolving_executor_test",
     timeout = "short",
     srcs = ["reference_resolving_executor_test.cc"],
@@ -609,7 +615,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "remote_executor",
     srcs = ["remote_executor.cc"],
     hdrs = ["remote_executor.h"],
@@ -631,7 +637,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "remote_executor_test",
     timeout = "short",
     srcs = ["remote_executor_test.cc"],
@@ -657,7 +663,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "sequence_executor",
     srcs = ["sequence_executor.cc"],
     hdrs = ["sequence_executor.h"],
@@ -683,7 +689,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "sequence_executor_test",
     timeout = "short",
     srcs = ["sequence_executor_test.cc"],
@@ -707,7 +713,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "sequence_intrinsics",
     srcs = ["sequence_intrinsics.cc"],
     hdrs = ["sequence_intrinsics.h"],
@@ -718,7 +724,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "session_provider",
     srcs = ["session_provider.cc"],
     hdrs = ["session_provider.h"],
@@ -740,7 +746,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "session_provider_test",
     srcs = ["session_provider_test.cc"],
     deps = [
@@ -751,7 +757,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "status_conversion",
     srcs = ["status_conversion.cc"],
     hdrs = ["status_conversion.h"],
@@ -761,7 +767,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "status_macros",
     hdrs = ["status_macros.h"],
     deps = [
@@ -771,7 +777,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "streaming_remote_executor",
     srcs = ["streaming_remote_executor.cc"],
     hdrs = ["streaming_remote_executor.h"],
@@ -796,7 +802,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "streaming_remote_executor_test",
     timeout = "short",
     srcs = ["streaming_remote_executor_test.cc"],
@@ -830,7 +836,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "struct_traversal_order",
     hdrs = ["struct_traversal_order.h"],
     deps = [
@@ -841,7 +847,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "struct_traversal_order_test",
     srcs = ["struct_traversal_order_test.cc"],
     deps = [
@@ -853,7 +859,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "tensor_serialization",
     srcs = ["tensor_serialization.cc"],
     hdrs = ["tensor_serialization.h"],
@@ -867,7 +873,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "tensorflow_executor",
     srcs = ["tensorflow_executor.cc"],
     hdrs = ["tensorflow_executor.h"],
@@ -943,7 +949,7 @@ tff_cc_cpu_gpu_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "tensorflow_utils",
     srcs = ["tensorflow_utils.cc"],
     hdrs = ["tensorflow_utils.h"],
@@ -960,7 +966,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "tensorflow_utils_test",
     srcs = ["tensorflow_utils_test.cc"],
     deps = [
@@ -980,7 +986,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "threading",
     srcs = ["threading.cc"],
     hdrs = ["threading.h"],
@@ -997,7 +1003,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "threading_test",
     timeout = "short",
     srcs = ["threading_test.cc"],
@@ -1012,7 +1018,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "type_test_utils",
     testonly = True,
     hdrs = ["type_test_utils.h"],
@@ -1023,7 +1029,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "type_utils",
     srcs = ["type_utils.cc"],
     hdrs = ["type_utils.h"],
@@ -1039,7 +1045,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "value_test_utils",
     testonly = True,
     hdrs = ["value_test_utils.h"],
@@ -1071,7 +1077,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "value_validation",
     srcs = ["value_validation.cc"],
     hdrs = ["value_validation.h"],
@@ -1086,7 +1092,7 @@ cc_library(
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "value_validation_test",
     timeout = "short",
     srcs = ["value_validation_test.cc"],
@@ -1102,7 +1108,7 @@ cc_test(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "xla_executor",
     srcs = ["xla_executor.cc"],
     hdrs = ["xla_executor.h"],
@@ -1118,20 +1124,20 @@ cc_library(
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
         "@com_google_absl//absl/types:span",
+        "@local_xla//xla:literal",
+        "@local_xla//xla:shape_util",
+        "@local_xla//xla:xla_data_proto_cc",
+        "@local_xla//xla:xla_proto_cc",
+        "@local_xla//xla/client",
+        "@local_xla//xla/client:client_library",
+        "@local_xla//xla/client:xla_computation",
+        "@local_xla//xla/service",
+        "@local_xla//xla/service:hlo_proto_cc",
+        "@local_xla//xla/stream_executor",
+        "@local_xla//xla/stream_executor:platform_manager",
+        "@local_xla//xla/stream_executor/host:host_platform",  # buildcleaner: keep # Linking in the host platform here ensures that the stream executor can execute on CPU.
         "@org_tensorflow//tensorflow/compiler/jit:xla_cpu_jit",  # buildcleaner: keep # Linking in this dependency ensures that XLA can compile its code for the CPU host.
         "@org_tensorflow//tensorflow/compiler/tf2xla:common",
-        "@org_tensorflow//tensorflow/compiler/xla:literal",
-        "@org_tensorflow//tensorflow/compiler/xla:shape_util",
-        "@org_tensorflow//tensorflow/compiler/xla:xla_data_proto_cc",
-        "@org_tensorflow//tensorflow/compiler/xla:xla_proto_cc",
-        "@org_tensorflow//tensorflow/compiler/xla/client",
-        "@org_tensorflow//tensorflow/compiler/xla/client:client_library",
-        "@org_tensorflow//tensorflow/compiler/xla/client:xla_computation",
-        "@org_tensorflow//tensorflow/compiler/xla/service",
-        "@org_tensorflow//tensorflow/compiler/xla/service:hlo_proto_cc",
-        "@org_tensorflow//tensorflow/compiler/xla/stream_executor",
-        "@org_tensorflow//tensorflow/compiler/xla/stream_executor:multi_platform_manager",
-        "@org_tensorflow//tensorflow/compiler/xla/stream_executor/host:host_platform",  # buildcleaner: keep # Linking in the host platform here ensures that the stream executor can execute on CPU.
         "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core:protos_all_cc",
     ],
@@ -1160,20 +1166,20 @@ tff_cc_cpu_gpu_test(
         "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
+        "@local_xla//xla:shape_util",
+        "@local_xla//xla:xla_data_proto_cc",
+        "@local_xla//xla/client:xla_builder",
+        "@local_xla//xla/client:xla_computation",
+        "@local_xla//xla/service:platform_util",
+        "@local_xla//xla/stream_executor",
         "@org_tensorflow//tensorflow/compiler/tf2xla:common",
-        "@org_tensorflow//tensorflow/compiler/xla:shape_util",
-        "@org_tensorflow//tensorflow/compiler/xla:xla_data_proto_cc",
-        "@org_tensorflow//tensorflow/compiler/xla/client:xla_builder",
-        "@org_tensorflow//tensorflow/compiler/xla/client:xla_computation",
-        "@org_tensorflow//tensorflow/compiler/xla/service:platform_util",
-        "@org_tensorflow//tensorflow/compiler/xla/stream_executor",
         "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core:protos_all_cc",
         "@org_tensorflow//tensorflow/core/platform:statusor",
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "xla_executor_gpu_test",
     srcs = ["xla_executor_test.cc"],
     args = ["--tff_xla_executor_test_platform=CUDA"],
@@ -1197,22 +1203,22 @@ cc_test(
         "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
+        "@local_xla//xla:shape_util",
+        "@local_xla//xla:xla_data_proto_cc",
+        "@local_xla//xla/client:xla_builder",
+        "@local_xla//xla/client:xla_computation",
+        "@local_xla//xla/service:platform_util",
+        "@local_xla//xla/stream_executor",
+        "@local_xla//xla/stream_executor/cuda:cuda_platform",  # buildcleaner: keep # Linking in the host platform here ensures that the stream executor can execute on GPU.
         "@org_tensorflow//tensorflow/compiler/jit:xla_gpu_jit",  # buildcleaner: keep # Linking in this dependency ensures that XLA can compile its code for the GPU.
         "@org_tensorflow//tensorflow/compiler/tf2xla:common",
-        "@org_tensorflow//tensorflow/compiler/xla:shape_util",
-        "@org_tensorflow//tensorflow/compiler/xla:xla_data_proto_cc",
-        "@org_tensorflow//tensorflow/compiler/xla/client:xla_builder",
-        "@org_tensorflow//tensorflow/compiler/xla/client:xla_computation",
-        "@org_tensorflow//tensorflow/compiler/xla/service:platform_util",
-        "@org_tensorflow//tensorflow/compiler/xla/stream_executor",
-        "@org_tensorflow//tensorflow/compiler/xla/stream_executor/cuda:cuda_platform",  # buildcleaner: keep # Linking in the host platform here ensures that the stream executor can execute on GPU.
         "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core:protos_all_cc",
         "@org_tensorflow//tensorflow/core/platform:statusor",
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "xla_utils",
     srcs = ["xla_utils.cc"],
     hdrs = ["xla_utils.h"],
@@ -1224,13 +1230,13 @@ cc_library(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
-        "@org_tensorflow//tensorflow/compiler/xla:literal",
-        "@org_tensorflow//tensorflow/compiler/xla:shape_util",
-        "@org_tensorflow//tensorflow/compiler/xla:types",
+        "@local_xla//xla:literal",
+        "@local_xla//xla:shape_util",
+        "@local_xla//xla:types",
     ],
 )
 
-cc_test(
+tff_cc_test(
     name = "xla_utils_test",
     srcs = ["xla_utils_test.cc"],
     deps = [
@@ -1245,10 +1251,10 @@ cc_test(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@eigen_archive//:eigen3",
-        "@org_tensorflow//tensorflow/compiler/xla:literal",
-        "@org_tensorflow//tensorflow/compiler/xla:literal_util",
-        "@org_tensorflow//tensorflow/compiler/xla:shape_util",
-        "@org_tensorflow//tensorflow/compiler/xla:types",
-        "@org_tensorflow//tensorflow/compiler/xla:xla_data_proto_cc",
+        "@local_xla//xla:literal",
+        "@local_xla//xla:literal_util",
+        "@local_xla//xla:shape_util",
+        "@local_xla//xla:types",
+        "@local_xla//xla:xla_data_proto_cc",
     ],
 )
diff --git a/tensorflow_federated/cc/core/impl/executors/array_test_utils.h b/tensorflow_federated/cc/core/impl/executors/array_test_utils.h
index 52bdf987d5..c018c2332a 100644
--- a/tensorflow_federated/cc/core/impl/executors/array_test_utils.h
+++ b/tensorflow_federated/cc/core/impl/executors/array_test_utils.h
@@ -23,7 +23,7 @@ limitations under the License
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
 #include "absl/strings/str_cat.h"
-#include "third_party/eigen3/Eigen/Core"
+#include "Eigen/Core"
 #include "tensorflow_federated/proto/v0/array.pb.h"
 #include "tensorflow_federated/proto/v0/data_type.pb.h"
 
diff --git a/tensorflow_federated/cc/core/impl/executors/tensorflow_utils.cc b/tensorflow_federated/cc/core/impl/executors/tensorflow_utils.cc
index d2f10a7fea..0b128f789d 100644
--- a/tensorflow_federated/cc/core/impl/executors/tensorflow_utils.cc
+++ b/tensorflow_federated/cc/core/impl/executors/tensorflow_utils.cc
@@ -24,7 +24,7 @@ limitations under the License
 #include "absl/status/statusor.h"
 #include "absl/strings/str_cat.h"
 #include "absl/strings/string_view.h"
-#include "third_party/eigen3/Eigen/Core"
+#include "Eigen/Core"
 #include "tensorflow/core/framework/attr_value.pb.h"
 #include "tensorflow/core/framework/graph.pb.h"
 #include "tensorflow/core/framework/node_def.pb.h"
diff --git a/tensorflow_federated/cc/core/impl/executors/tensorflow_utils_test.cc b/tensorflow_federated/cc/core/impl/executors/tensorflow_utils_test.cc
index 73d44bdcff..fd2e0b8465 100644
--- a/tensorflow_federated/cc/core/impl/executors/tensorflow_utils_test.cc
+++ b/tensorflow_federated/cc/core/impl/executors/tensorflow_utils_test.cc
@@ -22,7 +22,7 @@ limitations under the License
 #include "googletest/include/gtest/gtest.h"
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
-#include "third_party/eigen3/Eigen/Core"
+#include "Eigen/Core"
 #include "tensorflow/core/framework/numeric_types.h"
 #include "tensorflow/core/framework/tensor_shape.h"
 #include "tensorflow/core/framework/tensor_testutil.h"
diff --git a/tensorflow_federated/cc/core/impl/executors/xla_executor.cc b/tensorflow_federated/cc/core/impl/executors/xla_executor.cc
index 483b6e25ca..cff5d43be9 100644
--- a/tensorflow_federated/cc/core/impl/executors/xla_executor.cc
+++ b/tensorflow_federated/cc/core/impl/executors/xla_executor.cc
@@ -33,16 +33,17 @@ limitations under the License
 #include "absl/types/span.h"
 #include "tensorflow/compiler/tf2xla/literal_util.h"
 #include "tensorflow/compiler/tf2xla/type_util.h"
-#include "tensorflow/compiler/xla/client/client.h"
-#include "tensorflow/compiler/xla/client/client_library.h"
-#include "tensorflow/compiler/xla/client/xla_computation.h"
-#include "tensorflow/compiler/xla/literal.h"
-#include "tensorflow/compiler/xla/service/hlo.pb.h"
-#include "tensorflow/compiler/xla/service/service.h"
-#include "tensorflow/compiler/xla/shape.h"
-#include "tensorflow/compiler/xla/stream_executor/platform.h"
-#include "tensorflow/compiler/xla/xla.pb.h"
-#include "tensorflow/compiler/xla/xla_data.pb.h"
+#include "xla/client/client.h"
+#include "xla/client/client_library.h"
+#include "xla/client/xla_computation.h"
+#include "xla/literal.h"
+#include "xla/service/hlo.pb.h"
+#include "xla/service/service.h"
+#include "xla/shape.h"
+#include "xla/stream_executor/platform.h"
+#include "xla/stream_executor/platform_manager.h"
+#include "xla/xla.pb.h"
+#include "xla/xla_data.pb.h"
 #include "tensorflow/core/framework/tensor.h"
 #include "tensorflow/core/framework/tensor.pb.h"
 #include "tensorflow/core/framework/tensor_shape.h"
@@ -55,15 +56,6 @@ limitations under the License
 #include "tensorflow_federated/cc/core/impl/executors/xla_utils.h"
 #include "tensorflow_federated/proto/v0/computation.pb.h"
 
-// clang-format off
-// In TF 2.17 MultiPlatformManager was renamed to PlatformManager. Remove
-// this code when the OSS build gets updated to TF 2.17+.
-#include "tensorflow/compiler/xla/stream_executor/multi_platform_manager.h"
-namespace stream_executor {
-using PlatformManager = MultiPlatformManager;
-} // namespace stream_executor
-// clang-format on
-
 namespace tensorflow_federated {
 
 namespace {
@@ -696,8 +688,8 @@ class XLAExecutor : public ExecutorBase<ValueFuture> {
 };
 
 absl::StatusOr<xla::Client*> GetXLAClient(std::string_view platform_name) {
-  absl::StatusOr<xla::se::Platform*> platform =
-      xla::se::PlatformManager::PlatformWithName(platform_name);
+  absl::StatusOr<stream_executor::Platform*> platform =
+      stream_executor::PlatformManager::PlatformWithName(platform_name);
   if (!platform.ok()) {
     return absl::InternalError(
         absl::StrCat("Failed to find specified platform ", platform_name,
diff --git a/tensorflow_federated/cc/core/impl/executors/xla_executor_test.cc b/tensorflow_federated/cc/core/impl/executors/xla_executor_test.cc
index 7150bad17a..9b2cee8fed 100644
--- a/tensorflow_federated/cc/core/impl/executors/xla_executor_test.cc
+++ b/tensorflow_federated/cc/core/impl/executors/xla_executor_test.cc
@@ -32,13 +32,13 @@ limitations under the License
 #include "absl/status/statusor.h"
 #include "tensorflow/compiler/tf2xla/shape_util.h"
 #include "tensorflow/compiler/tf2xla/type_util.h"
-#include "tensorflow/compiler/xla/client/xla_builder.h"
-#include "tensorflow/compiler/xla/client/xla_computation.h"
-#include "tensorflow/compiler/xla/service/platform_util.h"
-#include "tensorflow/compiler/xla/shape.h"
-#include "tensorflow/compiler/xla/shape_util.h"
-#include "tensorflow/compiler/xla/stream_executor/platform.h"
-#include "tensorflow/compiler/xla/xla_data.pb.h"
+#include "xla/client/xla_builder.h"
+#include "xla/client/xla_computation.h"
+#include "xla/service/platform_util.h"
+#include "xla/shape.h"
+#include "xla/shape_util.h"
+#include "xla/stream_executor/platform.h"
+#include "xla/xla_data.pb.h"
 #include "tensorflow/core/framework/tensor_shape.h"
 #include "tensorflow/core/framework/types.pb.h"
 #include "tensorflow/core/platform/statusor.h"
diff --git a/tensorflow_federated/cc/core/impl/executors/xla_utils.cc b/tensorflow_federated/cc/core/impl/executors/xla_utils.cc
index ab94b79db9..c421f72bb7 100644
--- a/tensorflow_federated/cc/core/impl/executors/xla_utils.cc
+++ b/tensorflow_federated/cc/core/impl/executors/xla_utils.cc
@@ -22,10 +22,10 @@ limitations under the License
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
 #include "absl/strings/str_cat.h"
-#include "tensorflow/compiler/xla/literal.h"
-#include "tensorflow/compiler/xla/shape.h"
-#include "tensorflow/compiler/xla/shape_util.h"
-#include "tensorflow/compiler/xla/types.h"
+#include "xla/literal.h"
+#include "xla/shape.h"
+#include "xla/shape_util.h"
+#include "xla/types.h"
 #include "tensorflow_federated/cc/core/impl/executors/status_macros.h"
 #include "tensorflow_federated/proto/v0/array.pb.h"
 #include "tensorflow_federated/proto/v0/computation.pb.h"
diff --git a/tensorflow_federated/cc/core/impl/executors/xla_utils.h b/tensorflow_federated/cc/core/impl/executors/xla_utils.h
index 7a77b90059..80ea387a62 100644
--- a/tensorflow_federated/cc/core/impl/executors/xla_utils.h
+++ b/tensorflow_federated/cc/core/impl/executors/xla_utils.h
@@ -17,8 +17,8 @@ limitations under the License
 #define THIRD_PARTY_TENSORFLOW_FEDERATED_CC_CORE_IMPL_EXECUTORS_XLA_UTILS_H_
 
 #include "absl/status/statusor.h"
-#include "tensorflow/compiler/xla/literal.h"
-#include "tensorflow/compiler/xla/shape.h"
+#include "xla/literal.h"
+#include "xla/shape.h"
 #include "tensorflow_federated/proto/v0/array.pb.h"
 #include "tensorflow_federated/proto/v0/computation.pb.h"
 #include "tensorflow_federated/proto/v0/data_type.pb.h"
diff --git a/tensorflow_federated/cc/core/impl/executors/xla_utils_test.cc b/tensorflow_federated/cc/core/impl/executors/xla_utils_test.cc
index bab406f0e8..98c3befc04 100644
--- a/tensorflow_federated/cc/core/impl/executors/xla_utils_test.cc
+++ b/tensorflow_federated/cc/core/impl/executors/xla_utils_test.cc
@@ -21,14 +21,14 @@ limitations under the License
 #include "googletest/include/gtest/gtest.h"
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
-#include "third_party/eigen3/Eigen/Core"
-#include "tensorflow/compiler/xla/literal.h"
-#include "tensorflow/compiler/xla/literal_util.h"
-#include "tensorflow/compiler/xla/primitive_util.h"
-#include "tensorflow/compiler/xla/shape.h"
-#include "tensorflow/compiler/xla/shape_util.h"
-#include "tensorflow/compiler/xla/types.h"
-#include "tensorflow/compiler/xla/xla_data.pb.h"
+#include "Eigen/Core"
+#include "xla/literal.h"
+#include "xla/literal_util.h"
+#include "xla/primitive_util.h"
+#include "xla/shape.h"
+#include "xla/shape_util.h"
+#include "xla/types.h"
+#include "xla/xla_data.pb.h"
 #include "tensorflow_federated/cc/core/impl/executors/array_shape_test_utils.h"
 #include "tensorflow_federated/cc/core/impl/executors/array_test_utils.h"
 #include "tensorflow_federated/cc/testing/status_matchers.h"
diff --git a/tensorflow_federated/cc/simulation/BUILD b/tensorflow_federated/cc/simulation/BUILD
index aa5c601bd0..56ab3f5a34 100644
--- a/tensorflow_federated/cc/simulation/BUILD
+++ b/tensorflow_federated/cc/simulation/BUILD
@@ -1,3 +1,5 @@
+load("//tools:build_defs.bzl", "tff_cc_binary", "tff_cc_library")
+
 package(
     default_applicable_licenses = ["//:package_license"],
     default_visibility = [
@@ -23,7 +25,7 @@ package_group(
 
 licenses(["notice"])
 
-cc_library(
+tff_cc_library(
     name = "servers",
     srcs = ["servers.cc"],
     hdrs = ["servers.h"],
@@ -40,7 +42,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "worker_main",
     srcs = ["worker_main.cc"],
     deps = [
@@ -51,7 +53,7 @@ cc_library(
     ],
 )
 
-cc_binary(
+tff_cc_binary(
     name = "worker_binary",
     linkopts = ["-lrt"],
     deps = [":worker_main"],
diff --git a/tensorflow_federated/cc/testing/BUILD b/tensorflow_federated/cc/testing/BUILD
index 55c32c0e4d..22dc4a9d43 100644
--- a/tensorflow_federated/cc/testing/BUILD
+++ b/tensorflow_federated/cc/testing/BUILD
@@ -1,3 +1,5 @@
+load("//tools:build_defs.bzl", "tff_cc_library")
+
 package(
     default_applicable_licenses = ["//:package_license"],
     default_visibility = [
@@ -16,7 +18,7 @@ package_group(
 
 licenses(["notice"])
 
-cc_library(
+tff_cc_library(
     name = "oss_test_main",
     testonly = 1,
     srcs = ["oss_test_main.cc"],
@@ -27,7 +29,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "protobuf_matchers",
     testonly = True,
     hdrs = ["protobuf_matchers.h"],
@@ -39,7 +41,7 @@ cc_library(
     ],
 )
 
-cc_library(
+tff_cc_library(
     name = "status_matchers",
     testonly = True,
     srcs = ["status_matchers.cc"],
diff --git a/tensorflow_federated/python/aggregators/BUILD b/tensorflow_federated/python/aggregators/BUILD
index cec7274f8b..9fdaf7e695 100644
--- a/tensorflow_federated/python/aggregators/BUILD
+++ b/tensorflow_federated/python/aggregators/BUILD
@@ -55,6 +55,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -70,6 +71,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -89,6 +93,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -105,6 +110,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -123,6 +131,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -142,6 +152,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -160,6 +173,10 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dp_accounting//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -177,6 +194,10 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -194,6 +215,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -209,6 +232,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -235,6 +261,9 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -258,6 +287,10 @@ py_test(
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -276,6 +309,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -291,6 +326,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -326,18 +364,26 @@ py_test(
         ":sum_factory",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "hadamard",
     srcs = ["hadamard.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
     name = "hadamard_test",
     srcs = ["hadamard_test.py"],
-    deps = [":hadamard"],
+    deps = [
+        ":hadamard",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -355,6 +401,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -371,6 +419,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -402,6 +453,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -421,6 +474,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -435,6 +489,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -450,6 +507,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -465,6 +524,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -482,6 +544,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:estimation_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -495,6 +559,10 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -515,6 +583,8 @@ py_library(
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -536,6 +606,9 @@ py_test(
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -556,6 +629,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -573,6 +648,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -593,6 +671,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_transformations",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -605,6 +685,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -626,6 +709,8 @@ py_library(
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -646,6 +731,9 @@ py_test(
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -664,6 +752,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -683,6 +773,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -713,5 +806,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/analytics/BUILD b/tensorflow_federated/python/analytics/BUILD
index cefb441c82..a5e68de2dd 100644
--- a/tensorflow_federated/python/analytics/BUILD
+++ b/tensorflow_federated/python/analytics/BUILD
@@ -39,7 +39,10 @@ py_library(
 py_library(
     name = "data_processing",
     srcs = ["data_processing.py"],
-    deps = ["//tensorflow_federated/python/common_libs:py_typecheck"],
+    deps = [
+        "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -48,6 +51,8 @@ py_test(
     deps = [
         ":data_processing",
         ":histogram_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -55,11 +60,13 @@ py_library(
     name = "histogram_test_utils",
     testonly = True,
     srcs = ["histogram_test_utils.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_library(
     name = "histogram_processing",
     srcs = ["histogram_processing.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
@@ -68,6 +75,8 @@ py_test(
     deps = [
         ":histogram_processing",
         ":histogram_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -86,13 +95,18 @@ py_library(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_test(
     name = "differential_privacy_test",
     srcs = ["differential_privacy_test.py"],
-    deps = [":differential_privacy"],
+    deps = [
+        ":differential_privacy",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_test(
@@ -102,5 +116,8 @@ py_test(
         ":count_distinct",
         "//tensorflow_federated/python/core/backends/test:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/analytics/heavy_hitters/iblt/BUILD b/tensorflow_federated/python/analytics/heavy_hitters/iblt/BUILD
index 52a17dccb0..e1882d0322 100644
--- a/tensorflow_federated/python/analytics/heavy_hitters/iblt/BUILD
+++ b/tensorflow_federated/python/analytics/heavy_hitters/iblt/BUILD
@@ -45,6 +45,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -56,6 +58,9 @@ py_test(
         ":iblt_factory",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -65,18 +70,25 @@ py_library(
     deps = [
         ":chunkers",
         ":hyperedge_hashers",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "chunkers",
     srcs = ["chunkers.py"],
-    deps = ["//tensorflow_federated/python/common_libs:py_typecheck"],
+    deps = [
+        "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "hyperedge_hashers",
     srcs = ["hyperedge_hashers.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
@@ -85,20 +97,35 @@ py_test(
     timeout = "long",
     srcs = ["iblt_lib_test.py"],
     shard_count = 4,
-    deps = [":iblt_lib"],
+    deps = [
+        ":iblt_lib",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "chunkers_test",
     srcs = ["chunkers_test.py"],
-    deps = [":chunkers"],
+    deps = [
+        ":chunkers",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "hyperedge_hashers_test",
     size = "medium",
     srcs = ["hyperedge_hashers_test.py"],
-    deps = [":hyperedge_hashers"],
+    deps = [
+        ":hyperedge_hashers",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -116,6 +143,9 @@ py_library(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -131,6 +161,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -140,6 +173,8 @@ py_library(
     deps = [
         ":chunkers",
         ":iblt_lib",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -149,6 +184,8 @@ py_test(
     deps = [
         ":iblt_lib",
         ":iblt_tensor",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -168,6 +205,9 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -182,13 +222,20 @@ py_test(
         "//tensorflow_federated/python/aggregators:sum_factory",
         "//tensorflow_federated/python/core/backends/test:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "subsample_process",
     srcs = ["subsample_process.py"],
-    deps = [":iblt_factory"],
+    deps = [
+        ":iblt_factory",
+        "@pypi_scipy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -198,5 +245,8 @@ py_test(
         ":iblt_factory",
         ":iblt_lib",
         ":subsample_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/analytics/hierarchical_histogram/BUILD b/tensorflow_federated/python/analytics/hierarchical_histogram/BUILD
index 36be81313c..687cb9f68b 100644
--- a/tensorflow_federated/python/analytics/hierarchical_histogram/BUILD
+++ b/tensorflow_federated/python/analytics/hierarchical_histogram/BUILD
@@ -32,6 +32,7 @@ py_library(
         "//tensorflow_federated/python/aggregators:differential_privacy",
         "//tensorflow_federated/python/aggregators:secure",
         "//tensorflow_federated/python/aggregators:sum_factory",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -50,6 +51,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -65,6 +69,9 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:iterative_process",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -76,13 +83,20 @@ py_test(
         ":hierarchical_histogram_lib",
         "//tensorflow_federated/python/core/backends/test:execution_contexts",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "hierarchical_histogram_decoder",
     srcs = ["hierarchical_histogram_decoder.py"],
-    deps = [":build_tree_from_leaf"],
+    deps = [
+        ":build_tree_from_leaf",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -91,18 +105,26 @@ py_test(
     deps = [
         ":build_tree_from_leaf",
         ":hierarchical_histogram_decoder",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "build_tree_from_leaf",
     srcs = ["build_tree_from_leaf.py"],
+    deps = ["@pypi_numpy//:pkg"],
 )
 
 py_test(
     name = "build_tree_from_leaf_test",
     srcs = ["build_tree_from_leaf_test.py"],
-    deps = [":build_tree_from_leaf"],
+    deps = [
+        ":build_tree_from_leaf",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -118,6 +140,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/templates:aggregation_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -132,5 +156,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/common_libs/BUILD b/tensorflow_federated/python/common_libs/BUILD
index 06bdd7ae09..3fe6b96336 100644
--- a/tensorflow_federated/python/common_libs/BUILD
+++ b/tensorflow_federated/python/common_libs/BUILD
@@ -38,13 +38,17 @@ py_library(
 py_test(
     name = "deprecation_test",
     srcs = ["deprecation_test.py"],
-    deps = [":deprecation"],
+    deps = [
+        ":deprecation",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
     name = "golden",
     testonly = True,
     srcs = ["golden.py"],
+    deps = ["@pypi_absl_py//:pkg"],
 )
 
 py_test(
@@ -64,19 +68,29 @@ py_test(
         "golden_test_goldens/test_check_string_succeeds.expected",
         "golden_test_goldens/test_check_string_updates.expected",
     ],
-    deps = [":golden"],
+    deps = [
+        ":golden",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
     name = "py_typecheck",
     srcs = ["py_typecheck.py"],
+    deps = [
+        "@pypi_attrs//:pkg",
+        "@pypi_typing_extensions//:pkg",
+    ],
 )
 
 py_test(
     name = "py_typecheck_test",
     size = "small",
     srcs = ["py_typecheck_test.py"],
-    deps = [":py_typecheck"],
+    deps = [
+        ":py_typecheck",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
@@ -91,7 +105,10 @@ py_test(
     name = "retrying_test",
     size = "small",
     srcs = ["retrying_test.py"],
-    deps = [":retrying"],
+    deps = [
+        ":retrying",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
@@ -102,25 +119,39 @@ py_library(
 py_library(
     name = "structure",
     srcs = ["structure.py"],
-    deps = [":py_typecheck"],
+    deps = [
+        ":py_typecheck",
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
+    ],
 )
 
 py_test(
     name = "structure_test",
     size = "small",
     srcs = ["structure_test.py"],
-    deps = [":structure"],
+    deps = [
+        ":structure",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+    ],
 )
 
 py_library(
     name = "tracing",
     srcs = ["tracing.py"],
-    deps = [":py_typecheck"],
+    deps = [
+        ":py_typecheck",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_test(
     name = "tracing_test",
     size = "small",
     srcs = ["tracing_test.py"],
-    deps = [":tracing"],
+    deps = [
+        ":tracing",
+        "@pypi_absl_py//:pkg",
+    ],
 )
diff --git a/tensorflow_federated/python/core/backends/mapreduce/BUILD b/tensorflow_federated/python/core/backends/mapreduce/BUILD
index 5eb70ceaab..624c1f541c 100644
--- a/tensorflow_federated/python/core/backends/mapreduce/BUILD
+++ b/tensorflow_federated/python/core/backends/mapreduce/BUILD
@@ -52,6 +52,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -80,6 +82,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -112,6 +117,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -139,6 +147,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -166,6 +175,10 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/core/templates:iterative_process",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -185,6 +198,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/federated_context:value_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -207,6 +222,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -232,6 +248,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:value_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -251,5 +269,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:iterative_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/backends/native/BUILD b/tensorflow_federated/python/core/backends/native/BUILD
index 0efa6d2544..178b289801 100644
--- a/tensorflow_federated/python/core/backends/native/BUILD
+++ b/tensorflow_federated/python/core/backends/native/BUILD
@@ -41,6 +41,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/compiler:building_blocks",
         "//tensorflow_federated/python/core/impl/compiler:transformations",
         "//tensorflow_federated/python/core/impl/computation:computation_impl",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -55,6 +57,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -93,6 +97,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -121,6 +128,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/computation:computation_impl",
         "//tensorflow_federated/python/core/impl/executors:remote_executor_grpc_stub",
         "//tensorflow_federated/python/tensorflow_libs:tensorflow_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -160,5 +170,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/backends/test/BUILD b/tensorflow_federated/python/core/backends/test/BUILD
index 071ef0d838..9d859c8425 100644
--- a/tensorflow_federated/python/core/backends/test/BUILD
+++ b/tensorflow_federated/python/core/backends/test/BUILD
@@ -45,6 +45,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -61,6 +63,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -79,6 +83,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/execution_contexts:sync_execution_context",
         "//tensorflow_federated/python/core/impl/executor_stacks:cpp_executor_factory",
         "//tensorflow_federated/python/core/impl/executors:executor_bindings",
+        "@pypi_absl_py//:pkg",
+        "@pypi_portpicker//:pkg",
     ],
 )
 
@@ -98,6 +104,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -124,5 +133,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/backends/xla/BUILD b/tensorflow_federated/python/core/backends/xla/BUILD
index 488afec6c2..7cf62b5bcf 100644
--- a/tensorflow_federated/python/core/backends/xla/BUILD
+++ b/tensorflow_federated/python/core/backends/xla/BUILD
@@ -53,5 +53,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/environments/jax_frontend/BUILD b/tensorflow_federated/python/core/environments/jax_frontend/BUILD
index d3c3bf102c..0a92e2ee6c 100644
--- a/tensorflow_federated/python/core/environments/jax_frontend/BUILD
+++ b/tensorflow_federated/python/core/environments/jax_frontend/BUILD
@@ -33,6 +33,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
+        "//third_party/jax",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -45,6 +47,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/computation:computation_impl",
         "//tensorflow_federated/python/core/impl/computation:polymorphic_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "//third_party/jax",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -61,6 +66,8 @@ py_test(
     deps = [
         ":jax_computation",
         ":jax_computation_context",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -79,6 +86,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:typed_object",
+        "//third_party/jax",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -95,5 +104,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "//third_party/jax",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/environments/tensorflow_backend/BUILD b/tensorflow_federated/python/core/environments/tensorflow_backend/BUILD
index 18265f4b99..114111cfc2 100644
--- a/tensorflow_federated/python/core/environments/tensorflow_backend/BUILD
+++ b/tensorflow_federated/python/core/environments/tensorflow_backend/BUILD
@@ -51,13 +51,19 @@ py_test(
         ":tensorflow_computation_transformations",
         "//tensorflow_federated/python/core/impl/compiler:building_blocks",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "graph_optimizations",
     srcs = ["graph_optimizations.py"],
-    deps = [":graph_spec"],
+    deps = [
+        ":graph_spec",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_test(
@@ -66,18 +72,26 @@ py_test(
     deps = [
         ":graph_optimizations",
         ":graph_spec",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "graph_spec",
     srcs = ["graph_spec.py"],
+    deps = [
+        "@pypi_attrs//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "graph_spec_test",
     srcs = ["graph_spec_test.py"],
-    deps = [":graph_spec"],
+    deps = [
+        ":graph_spec",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -108,6 +122,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/compiler:intrinsic_defs",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -127,6 +143,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_transformations",
         "//tensorflow_federated/python/core/impl/utils:tensorflow_utils",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -142,6 +160,9 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_factory",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -155,6 +176,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/utils:tensorflow_utils",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -165,6 +188,7 @@ py_library(
         "//tensorflow_federated/proto/v0:computation_py_pb2",
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -181,6 +205,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/utils:tensorflow_utils",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -190,7 +217,10 @@ py_library(
     tags = [
         "nokokoro",  # b/193543632: C++ execution is not fully supported in OSS.
     ],
-    deps = ["//tensorflow_federated/cc/core/impl/executors:executor_bindings"],
+    deps = [
+        "//tensorflow_federated/cc/core/impl/executors:executor_bindings",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -208,6 +238,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -225,6 +258,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -239,5 +274,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/environments/tensorflow_frontend/BUILD b/tensorflow_federated/python/core/environments/tensorflow_frontend/BUILD
index a30279b6de..22b4bd6595 100644
--- a/tensorflow_federated/python/core/environments/tensorflow_frontend/BUILD
+++ b/tensorflow_federated/python/core/environments/tensorflow_frontend/BUILD
@@ -35,6 +35,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -49,6 +51,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:runtime_error_context",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -64,6 +69,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/impl/utils:tensorflow_utils",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -79,6 +86,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -96,6 +106,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/utils:tensorflow_utils",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -110,12 +121,19 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
         "//tensorflow_federated/python/tensorflow_libs:tensorflow_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "variable_utils",
     srcs = ["variable_utils.py"],
+    deps = [
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -123,5 +141,9 @@ py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["variable_utils_test.py"],
-    deps = [":variable_utils"],
+    deps = [
+        ":variable_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
diff --git a/tensorflow_federated/python/core/environments/xla_backend/BUILD b/tensorflow_federated/python/core/environments/xla_backend/BUILD
index 7385743398..ecdd5b7b83 100644
--- a/tensorflow_federated/python/core/environments/xla_backend/BUILD
+++ b/tensorflow_federated/python/core/environments/xla_backend/BUILD
@@ -33,6 +33,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/compiler:local_computation_factory_base",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
+        "//third_party/jax",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -45,6 +47,9 @@ py_test(
         ":runtime",
         "//tensorflow_federated/proto/v0:computation_py_pb2",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "//third_party/jax",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -60,6 +65,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:typed_object",
+        "//third_party/jax",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -71,6 +78,9 @@ py_test(
         ":xla_serialization",
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "//third_party/jax",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -80,7 +90,10 @@ py_library(
     tags = [
         "nokokoro",  # b/193543632: C++ execution is not fully supported in OSS.
     ],
-    deps = ["//tensorflow_federated/cc/core/impl/executors:executor_bindings"],
+    deps = [
+        "//tensorflow_federated/cc/core/impl/executors:executor_bindings",
+        "@pypi_tensorflow//:pkg",  # TODO: b/305743962 - Remove the tensorflow dependency.
+    ],
 )
 
 py_test(
@@ -89,7 +102,10 @@ py_test(
     tags = [
         "nokokoro",  # b/193543632: C++ execution is not fully supported in OSS.
     ],
-    deps = [":xla_executor_bindings"],
+    deps = [
+        ":xla_executor_bindings",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
@@ -101,7 +117,9 @@ py_library(
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "//third_party/jax",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -114,6 +132,9 @@ py_test(
         "//tensorflow_federated/proto/v0:computation_py_pb2",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "//third_party/jax",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/impl/compiler/BUILD b/tensorflow_federated/python/core/impl/compiler/BUILD
index 9ed75b1f48..ff6f2a4fa9 100644
--- a/tensorflow_federated/python/core/impl/compiler/BUILD
+++ b/tensorflow_federated/python/core/impl/compiler/BUILD
@@ -32,6 +32,7 @@ py_library(
         "//tensorflow_federated/proto/v0:array_py_pb2",
         "//tensorflow_federated/python/core/impl/types:array_shape",
         "//tensorflow_federated/python/core/impl/types:dtype_utils",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -42,6 +43,8 @@ py_test(
         ":array",
         "//tensorflow_federated/proto/v0:array_py_pb2",
         "//tensorflow_federated/proto/v0:data_type_py_pb2",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -122,6 +125,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -136,6 +141,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -154,6 +160,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/types:typed_object",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -175,6 +182,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -198,6 +208,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_factory",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -209,6 +221,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_factory",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -216,7 +229,10 @@ py_test(
     name = "intrinsic_defs_test",
     size = "small",
     srcs = ["intrinsic_defs_test.py"],
-    deps = [":intrinsic_defs"],
+    deps = [
+        ":intrinsic_defs",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
@@ -241,6 +257,7 @@ py_library(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_attrs//:pkg",
     ],
 )
 
@@ -262,6 +279,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -288,6 +307,8 @@ py_test(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -317,6 +338,8 @@ py_test(
         ":tree_analysis",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -362,6 +385,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -391,5 +416,6 @@ py_test(
     deps = [
         ":building_block_test_utils",
         ":compiler_test_utils",
+        "@pypi_absl_py//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/impl/computation/BUILD b/tensorflow_federated/python/core/impl/computation/BUILD
index 3f27862385..55eaedeef5 100644
--- a/tensorflow_federated/python/core/impl/computation/BUILD
+++ b/tensorflow_federated/python/core/impl/computation/BUILD
@@ -61,6 +61,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -87,6 +89,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/compiler:computation_factory",
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -117,6 +121,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -141,6 +147,8 @@ py_test(
         ":function_utils",
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -167,5 +175,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/impl/context_stack/BUILD b/tensorflow_federated/python/core/impl/context_stack/BUILD
index 8ced9f4a1c..1ed7da969b 100644
--- a/tensorflow_federated/python/core/impl/context_stack/BUILD
+++ b/tensorflow_federated/python/core/impl/context_stack/BUILD
@@ -53,6 +53,7 @@ py_test(
     deps = [
         ":context_stack_impl",
         ":context_stack_test_utils",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -62,6 +63,7 @@ py_library(
     deps = [
         ":context_base",
         ":context_stack_impl",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -71,6 +73,7 @@ py_test(
     deps = [
         ":context_stack_impl",
         ":context_stack_test_utils",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -87,6 +90,7 @@ py_test(
     deps = [
         ":context_stack_impl",
         ":get_context_stack",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -113,6 +117,7 @@ py_test(
         ":context_stack_impl",
         ":context_stack_test_utils",
         ":set_default_context",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
diff --git a/tensorflow_federated/python/core/impl/execution_contexts/BUILD b/tensorflow_federated/python/core/impl/execution_contexts/BUILD
index 643296ead3..0be678ce41 100644
--- a/tensorflow_federated/python/core/impl/execution_contexts/BUILD
+++ b/tensorflow_federated/python/core/impl/execution_contexts/BUILD
@@ -51,6 +51,7 @@ py_test(
     deps = [
         ":async_execution_context",
         "//tensorflow_federated/python/core/impl/executors:executors_errors",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -70,6 +71,7 @@ py_test(
     deps = [
         ":compiler_pipeline",
         "//tensorflow_federated/python/core/impl/computation:computation_base",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -91,6 +93,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/impl/types:typed_object",
+        "@pypi_attrs//:pkg",
     ],
 )
 
@@ -104,6 +107,8 @@ py_test(
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
diff --git a/tensorflow_federated/python/core/impl/executor_stacks/BUILD b/tensorflow_federated/python/core/impl/executor_stacks/BUILD
index a97fcfc5e8..1b4fda3afa 100644
--- a/tensorflow_federated/python/core/impl/executor_stacks/BUILD
+++ b/tensorflow_federated/python/core/impl/executor_stacks/BUILD
@@ -39,6 +39,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/executors:executor_factory",
         "//tensorflow_federated/python/core/impl/executors:executors_errors",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_cachetools//:pkg",
     ],
 )
 
@@ -55,6 +57,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/executors:executor_factory",
         "//tensorflow_federated/python/core/impl/executors:executor_test_utils_bindings",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -69,6 +72,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/executors:remote_executor",
         "//tensorflow_federated/python/core/impl/executors:remote_executor_grpc_stub",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_portpicker//:pkg",
     ],
 )
 
@@ -97,6 +102,8 @@ py_test(
         ":executor_stack_bindings",
         "//tensorflow_federated/python/core/impl/executors:executor_bindings",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -108,6 +115,7 @@ py_library(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/core/impl/executors:executor_base",
         "//tensorflow_federated/python/core/impl/executors:executor_factory",
+        "@pypi_cachetools//:pkg",
     ],
 )
 
@@ -121,5 +129,6 @@ py_test(
         "//tensorflow_federated/python/core/impl/executors:executor_base",
         "//tensorflow_federated/python/core/impl/executors:executor_factory",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/impl/executors/BUILD b/tensorflow_federated/python/core/impl/executors/BUILD
index a59fb8c107..df1f8977c4 100644
--- a/tensorflow_federated/python/core/impl/executors/BUILD
+++ b/tensorflow_federated/python/core/impl/executors/BUILD
@@ -45,6 +45,8 @@ py_test(
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -84,6 +86,8 @@ py_test(
         ":value_serialization",
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -100,6 +104,7 @@ py_test(
     deps = [
         ":data_conversions",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -127,6 +132,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -148,6 +155,7 @@ py_library(
         ":data_conversions",
         "//tensorflow_federated/cc/core/impl/executors:executor_bindings",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -162,6 +170,11 @@ py_test(
         ":executor_bindings",
         ":executor_test_utils_bindings",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_portpicker//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -203,6 +216,8 @@ py_test(
         ":executor_utils",
         "//tensorflow_federated/python/core/impl/compiler:building_block_factory",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -217,6 +232,7 @@ py_library(
     name = "executors_errors",
     srcs = ["executors_errors.py"],
     srcs_version = "PY3",
+    deps = ["@pypi_typing_extensions//:pkg"],
 )
 
 py_library(
@@ -242,6 +258,7 @@ py_library(
         "//tensorflow_federated/python/common_libs:tracing",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -257,6 +274,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/compiler:computation_factory",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -277,6 +296,7 @@ py_library(
         "//tensorflow_federated/proto/v0:executor_py_pb2",
         "//tensorflow_federated/proto/v0:executor_py_pb2_grpc",
         "//tensorflow_federated/python/common_libs:tracing",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -290,6 +310,8 @@ py_test(
         "//tensorflow_federated/proto/v0:executor_py_pb2",
         "//tensorflow_federated/proto/v0:executor_py_pb2_grpc",
         "//tensorflow_federated/python/core/impl/compiler:computation_factory",
+        "@pypi_absl_py//:pkg",
+        "@pypi_portpicker//:pkg",
     ],
 )
 
@@ -315,6 +337,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_transformations",
         "//tensorflow_federated/python/core/impl/utils:tensorflow_utils",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -334,5 +358,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/impl/federated_context/BUILD b/tensorflow_federated/python/core/impl/federated_context/BUILD
index 720cb4c4d1..c143fea5fd 100644
--- a/tensorflow_federated/python/core/impl/federated_context/BUILD
+++ b/tensorflow_federated/python/core/impl/federated_context/BUILD
@@ -47,6 +47,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:get_context_stack",
         "//tensorflow_federated/python/core/impl/context_stack:runtime_error_context",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -78,6 +80,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -106,6 +110,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/computation:function_utils",
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -128,6 +134,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_analysis",
         "//tensorflow_federated/python/core/impl/types:type_factory",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -144,6 +151,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_test_utils",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -166,6 +175,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/impl/types:typed_object",
+        "@pypi_attrs//:pkg",
     ],
 )
 
@@ -183,6 +193,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -213,5 +226,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/impl/types/BUILD b/tensorflow_federated/python/core/impl/types/BUILD
index 181f37daa9..3523169378 100644
--- a/tensorflow_federated/python/core/impl/types/BUILD
+++ b/tensorflow_federated/python/core/impl/types/BUILD
@@ -50,6 +50,7 @@ py_test(
     deps = [
         ":array_shape",
         "//tensorflow_federated/proto/v0:array_py_pb2",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -61,6 +62,11 @@ py_library(
         ":placements",
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_typing_extensions//:pkg",
     ],
 )
 
@@ -86,19 +92,30 @@ py_test(
         ":placements",
         "//tensorflow_federated/python/common_libs:golden",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "dtype_utils",
     srcs = ["dtype_utils.py"],
-    deps = ["//tensorflow_federated/proto/v0:data_type_py_pb2"],
+    deps = [
+        "//tensorflow_federated/proto/v0:data_type_py_pb2",
+        "@pypi_numpy//:pkg",
+    ],
 )
 
 py_test(
     name = "dtype_utils_test",
     srcs = ["dtype_utils_test.py"],
-    deps = [":dtype_utils"],
+    deps = [
+        ":dtype_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+    ],
 )
 
 py_library(
@@ -110,7 +127,10 @@ py_test(
     name = "placements_test",
     size = "small",
     srcs = ["placements_test.py"],
-    deps = [":placements"],
+    deps = [
+        ":placements",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
@@ -124,6 +144,7 @@ py_library(
         ":type_transformations",
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -136,6 +157,8 @@ py_test(
         ":placements",
         ":type_analysis",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -148,6 +171,10 @@ py_library(
         ":typed_object",
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -162,6 +189,10 @@ py_test(
         ":type_test_utils",
         ":typed_object",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -178,6 +209,8 @@ py_test(
     deps = [
         ":computation_types",
         ":type_factory",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -207,6 +240,8 @@ py_test(
         ":type_serialization",
         "//tensorflow_federated/proto/v0:computation_py_pb2",
         "//tensorflow_federated/proto/v0:data_type_py_pb2",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -234,6 +269,8 @@ py_test(
         ":computation_types",
         ":placements",
         ":type_transformations",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
diff --git a/tensorflow_federated/python/core/impl/utils/BUILD b/tensorflow_federated/python/core/impl/utils/BUILD
index 29cc3fa7f9..215fb4c488 100644
--- a/tensorflow_federated/python/core/impl/utils/BUILD
+++ b/tensorflow_federated/python/core/impl/utils/BUILD
@@ -37,6 +37,9 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_serialization",
         "//tensorflow_federated/python/tensorflow_libs:graph_utils",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -53,5 +56,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
         "//tensorflow_federated/python/tensorflow_libs:serialization_utils",
         "//tensorflow_federated/python/tensorflow_libs:tensorflow_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/templates/BUILD b/tensorflow_federated/python/core/templates/BUILD
index 4f00359cd0..5c372d1934 100644
--- a/tensorflow_federated/python/core/templates/BUILD
+++ b/tensorflow_federated/python/core/templates/BUILD
@@ -53,6 +53,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -84,6 +86,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -110,6 +114,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -124,6 +130,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
     ],
 )
 
@@ -158,5 +166,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/core/test/BUILD b/tensorflow_federated/python/core/test/BUILD
index 259c6cad36..9d74d8ee4e 100644
--- a/tensorflow_federated/python/core/test/BUILD
+++ b/tensorflow_federated/python/core/test/BUILD
@@ -48,5 +48,6 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:federated_computation",
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/learning/BUILD b/tensorflow_federated/python/learning/BUILD
index 92498e5d1f..1e43dd7211 100644
--- a/tensorflow_federated/python/learning/BUILD
+++ b/tensorflow_federated/python/learning/BUILD
@@ -44,17 +44,24 @@ py_library(
 py_library(
     name = "client_weight_lib",
     srcs = ["client_weight_lib.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_library(
     name = "loop_builder",
     srcs = ["loop_builder.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_cpu_gpu_test(
     name = "loop_builder_test",
     srcs = ["loop_builder_test.py"],
-    deps = [":loop_builder"],
+    deps = [
+        ":loop_builder",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -89,6 +96,8 @@ py_test(
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:iterative_process",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -101,6 +110,7 @@ py_library(
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -114,18 +124,28 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:federated_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "tensor_utils",
     srcs = ["tensor_utils.py"],
-    deps = ["//tensorflow_federated/python/common_libs:py_typecheck"],
+    deps = [
+        "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "tensor_utils_test",
     size = "small",
     srcs = ["tensor_utils_test.py"],
-    deps = [":tensor_utils"],
+    deps = [
+        ":tensor_utils",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
diff --git a/tensorflow_federated/python/learning/algorithms/BUILD b/tensorflow_federated/python/learning/algorithms/BUILD
index 174a8eb725..c8c6f403e6 100644
--- a/tensorflow_federated/python/learning/algorithms/BUILD
+++ b/tensorflow_federated/python/learning/algorithms/BUILD
@@ -58,6 +58,8 @@ py_library(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
         "//tensorflow_federated/python/learning/templates:model_delta_client_work",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -76,6 +78,8 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/models:model_examples",
         "//tensorflow_federated/python/learning/models:test_models",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -107,6 +111,8 @@ py_library(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
         "//tensorflow_federated/python/learning/templates:model_delta_client_work",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -124,6 +130,8 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/models:model_examples",
         "//tensorflow_federated/python/learning/models:test_models",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -150,6 +158,9 @@ py_library(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
         "//tensorflow_federated/python/learning/templates:proximal_client_work",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -171,6 +182,8 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/models:test_models",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
         "//tensorflow_federated/python/learning/templates:distributors",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -201,6 +214,8 @@ py_library(
         "//tensorflow_federated/python/learning/templates:composers",
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -232,6 +247,11 @@ py_test(
         "//tensorflow_federated/python/learning/optimizers:sgdm",
         "//tensorflow_federated/python/learning/templates:composers",
         "//tensorflow_federated/python/learning/templates:distributors",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tensorflow_privacy//:pkg",
     ],
 )
 
@@ -258,6 +278,7 @@ py_library(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:finalizers",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -280,6 +301,10 @@ py_test(
         "//tensorflow_federated/python/learning/templates:composers",
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_attrs//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -309,6 +334,8 @@ py_library(
         "//tensorflow_federated/python/learning/templates:composers",
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -328,6 +355,9 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/models:test_models",
         "//tensorflow_federated/python/tensorflow_libs:tensorflow_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -349,6 +379,8 @@ py_library(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:finalizers",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -360,6 +392,9 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -393,6 +428,8 @@ py_library(
         "//tensorflow_federated/python/learning/templates:composers",
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -436,6 +473,9 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/templates:client_works",
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/tensorflow_libs:tensorflow_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -463,6 +503,7 @@ py_library(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:finalizers",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -493,6 +534,9 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/templates:distributors",
         "//tensorflow_federated/python/learning/templates:learning_process",
         "//tensorflow_federated/python/tensorflow_libs:tensorflow_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -512,6 +556,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/models:variable",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -527,5 +572,8 @@ py_cpu_gpu_test(
         "//tensorflow_federated/python/learning/models:keras_utils",
         "//tensorflow_federated/python/learning/models:model_examples",
         "//tensorflow_federated/python/learning/models:model_weights",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/learning/metrics/BUILD b/tensorflow_federated/python/learning/metrics/BUILD
index 3cffc7bf16..f6c94ede34 100644
--- a/tensorflow_federated/python/learning/metrics/BUILD
+++ b/tensorflow_federated/python/learning/metrics/BUILD
@@ -53,6 +53,9 @@ py_library(
         "//tensorflow_federated/python/core/templates:aggregation_process",
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -71,6 +74,9 @@ py_test(
         "//tensorflow_federated/python/core/templates:estimation_process",
         "//tensorflow_federated/python/core/templates:measured_process",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -95,6 +101,9 @@ py_test(
         ":keras_finalizer",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -128,24 +137,39 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/test:static_assert",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "counters",
     srcs = ["counters.py"],
+    deps = [
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "counters_test",
     srcs = ["counters_test.py"],
-    deps = [":counters"],
+    deps = [
+        ":counters",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "keras_finalizer",
     srcs = ["keras_finalizer.py"],
-    deps = ["//tensorflow_federated/python/common_libs:py_typecheck"],
+    deps = [
+        "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -156,13 +180,19 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "keras_utils",
     srcs = ["keras_utils.py"],
-    deps = ["//tensorflow_federated/python/core/environments/tensorflow_frontend:variable_utils"],
+    deps = [
+        "//tensorflow_federated/python/core/environments/tensorflow_frontend:variable_utils",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -171,6 +201,9 @@ py_test(
     deps = [
         ":counters",
         ":keras_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -183,6 +216,9 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
diff --git a/tensorflow_federated/python/learning/models/BUILD b/tensorflow_federated/python/learning/models/BUILD
index bbd7fd48a3..72a46b14ad 100644
--- a/tensorflow_federated/python/learning/models/BUILD
+++ b/tensorflow_federated/python/learning/models/BUILD
@@ -26,6 +26,8 @@ py_library(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -53,6 +55,8 @@ py_library(
         "//tensorflow_federated/python/learning/metrics:keras_finalizer",
         "//tensorflow_federated/python/learning/metrics:keras_utils",
         "//tensorflow_federated/python/learning/metrics:types",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -69,6 +73,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/learning/metrics:aggregator",
         "//tensorflow_federated/python/learning/metrics:types",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -85,6 +91,8 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/learning/metrics:counters",
         "//tensorflow_federated/python/learning/metrics:keras_finalizer",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -105,13 +113,19 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/learning/metrics:aggregator",
         "//tensorflow_federated/python/learning/metrics:counters",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "model_examples",
     srcs = ["model_examples.py"],
-    deps = [":variable"],
+    deps = [
+        ":variable",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -122,6 +136,8 @@ py_test(
         ":model_examples",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -139,6 +155,7 @@ py_library(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_conversions",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -150,6 +167,9 @@ py_test(
         ":variable",
         "//tensorflow_federated/python/common_libs:structure",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -160,6 +180,7 @@ py_library(
         ":model_weights",
         ":variable",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -170,6 +191,9 @@ py_test(
         ":model_weights",
         ":reconstruction_model",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -187,6 +211,9 @@ py_test(
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -198,11 +225,16 @@ py_library(
         ":functional",
         ":variable",
         "//tensorflow_federated/python/learning/metrics:types",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "variable",
     srcs = ["variable.py"],
-    deps = ["//tensorflow_federated/python/learning/metrics:types"],
+    deps = [
+        "//tensorflow_federated/python/learning/metrics:types",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
diff --git a/tensorflow_federated/python/learning/optimizers/BUILD b/tensorflow_federated/python/learning/optimizers/BUILD
index b778105f0e..b355726d05 100644
--- a/tensorflow_federated/python/learning/optimizers/BUILD
+++ b/tensorflow_federated/python/learning/optimizers/BUILD
@@ -23,6 +23,7 @@ py_library(
     deps = [
         ":optimizer",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -33,6 +34,8 @@ py_test(
         ":adafactor",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -42,6 +45,7 @@ py_library(
     deps = [
         ":optimizer",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -52,6 +56,8 @@ py_test(
         ":adagrad",
         ":optimizer",
         ":optimizer_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -61,6 +67,7 @@ py_library(
     deps = [
         ":optimizer",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -71,6 +78,8 @@ py_test(
         ":adam",
         ":optimizer",
         ":optimizer_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -93,7 +102,10 @@ py_library(
 py_library(
     name = "keras_optimizer",
     srcs = ["keras_optimizer.py"],
-    deps = [":optimizer"],
+    deps = [
+        ":optimizer",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -106,6 +118,8 @@ py_test(
         ":sgdm",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -125,31 +139,48 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:intrinsics",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "optimizer",
     srcs = ["optimizer.py"],
+    deps = [
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "optimizer_test",
     srcs = ["optimizer_test.py"],
-    deps = [":optimizer"],
+    deps = [
+        ":optimizer",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "optimizer_test_utils",
     testonly = True,
     srcs = ["optimizer_test_utils.py"],
-    deps = [":optimizer"],
+    deps = [
+        ":optimizer",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "optimizer_test_utils_test",
     srcs = ["optimizer_test_utils_test.py"],
-    deps = [":optimizer_test_utils"],
+    deps = [
+        ":optimizer_test_utils",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -158,6 +189,7 @@ py_library(
     deps = [
         ":optimizer",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -168,13 +200,18 @@ py_test(
         ":optimizer",
         ":optimizer_test_utils",
         ":rmsprop",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "scheduling",
     srcs = ["scheduling.py"],
-    deps = [":optimizer"],
+    deps = [
+        ":optimizer",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -188,6 +225,8 @@ py_test(
         ":scheduling",
         ":sgdm",
         ":yogi",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -197,6 +236,7 @@ py_library(
     deps = [
         ":optimizer",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -207,6 +247,8 @@ py_test(
         ":optimizer",
         ":optimizer_test_utils",
         ":sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -216,6 +258,7 @@ py_library(
     deps = [
         ":optimizer",
         "//tensorflow_federated/python/common_libs:structure",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -226,5 +269,7 @@ py_test(
         ":optimizer",
         ":optimizer_test_utils",
         ":yogi",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/learning/programs/BUILD b/tensorflow_federated/python/learning/programs/BUILD
index c74de42e98..3f640e09ca 100644
--- a/tensorflow_federated/python/learning/programs/BUILD
+++ b/tensorflow_federated/python/learning/programs/BUILD
@@ -35,6 +35,8 @@ py_library(
         "//tensorflow_federated/python/program:program_state_manager",
         "//tensorflow_federated/python/program:release_manager",
         "//tensorflow_federated/python/program:value_reference",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -59,6 +61,9 @@ py_test(
         "//tensorflow_federated/python/program:file_program_state_manager",
         "//tensorflow_federated/python/program:native_platform",
         "//tensorflow_federated/python/program:release_manager",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -87,6 +92,7 @@ py_library(
         "//tensorflow_federated/python/program:program_state_manager",
         "//tensorflow_federated/python/program:release_manager",
         "//tensorflow_federated/python/program:value_reference",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -108,6 +114,8 @@ py_test(
         "//tensorflow_federated/python/program:native_platform",
         "//tensorflow_federated/python/program:program_state_manager",
         "//tensorflow_federated/python/program:release_manager",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -124,6 +132,7 @@ py_library(
         "//tensorflow_federated/python/program:release_manager",
         "//tensorflow_federated/python/program:structure_utils",
         "//tensorflow_federated/python/program:value_reference",
+        "@pypi_google_vizier//:pkg",
     ],
 )
 
@@ -141,5 +150,7 @@ py_test(
         "//tensorflow_federated/python/program:native_platform",
         "//tensorflow_federated/python/program:program_state_manager",
         "//tensorflow_federated/python/program:release_manager",
+        "@pypi_absl_py//:pkg",
+        "@pypi_google_vizier//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/learning/templates/BUILD b/tensorflow_federated/python/learning/templates/BUILD
index 92709405d6..2b681a85cd 100644
--- a/tensorflow_federated/python/learning/templates/BUILD
+++ b/tensorflow_federated/python/learning/templates/BUILD
@@ -49,6 +49,7 @@ py_library(
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/optimizers:keras_optimizer",
         "//tensorflow_federated/python/learning/optimizers:optimizer",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -66,6 +67,9 @@ py_test(
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/optimizers:optimizer",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -99,6 +103,8 @@ py_test(
         "//tensorflow_federated/python/core/templates:errors",
         "//tensorflow_federated/python/core/templates:measured_process",
         "//tensorflow_federated/python/learning/models:model_weights",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -157,6 +163,9 @@ py_test(
         "//tensorflow_federated/python/learning/models:model_examples",
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -190,6 +199,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:errors",
         "//tensorflow_federated/python/core/templates:measured_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -224,6 +236,8 @@ py_test(
         "//tensorflow_federated/python/core/templates:errors",
         "//tensorflow_federated/python/core/templates:measured_process",
         "//tensorflow_federated/python/learning/models:model_weights",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -246,6 +260,8 @@ py_test(
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -275,6 +291,9 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:errors",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -299,6 +318,7 @@ py_library(
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/models:variable",
         "//tensorflow_federated/python/learning/optimizers:optimizer",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -326,6 +346,9 @@ py_test(
         "//tensorflow_federated/python/learning/models:model_examples",
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -350,6 +373,8 @@ py_library(
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/models:variable",
         "//tensorflow_federated/python/learning/optimizers:optimizer",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -375,6 +400,9 @@ py_test(
         "//tensorflow_federated/python/learning/models:model_weights",
         "//tensorflow_federated/python/learning/models:test_models",
         "//tensorflow_federated/python/learning/optimizers:sgdm",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -395,5 +423,7 @@ py_test(
         ":type_checks",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/program/BUILD b/tensorflow_federated/python/program/BUILD
index e8d53e1561..0163ad7d48 100644
--- a/tensorflow_federated/python/program/BUILD
+++ b/tensorflow_federated/python/program/BUILD
@@ -47,6 +47,7 @@ py_library(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -57,6 +58,8 @@ py_test(
         ":client_id_data_source",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -78,6 +81,7 @@ py_library(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -88,6 +92,9 @@ py_test(
         ":dataset_data_source",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -116,6 +123,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/context_stack:context_stack_impl",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -129,6 +138,9 @@ py_library(
         ":value_reference",
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/common_libs:serializable",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -140,6 +152,10 @@ py_test(
         ":file_utils",
         ":program_state_manager",
         ":program_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -152,6 +168,8 @@ py_library(
         ":structure_utils",
         ":value_reference",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -164,6 +182,10 @@ py_test(
         ":program_test_utils",
         ":release_manager",
         ":structure_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -173,13 +195,19 @@ py_library(
     deps = [
         ":structure_utils",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_test(
     name = "file_utils_test",
     srcs = ["file_utils_test.py"],
-    deps = [":file_utils"],
+    deps = [
+        ":file_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -188,6 +216,7 @@ py_library(
     deps = [
         ":release_manager",
         ":value_reference",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -197,6 +226,10 @@ py_test(
     deps = [
         ":logging_release_manager",
         ":program_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -216,6 +249,10 @@ py_test(
     deps = [
         ":memory_release_manager",
         ":program_test_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -251,6 +288,10 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:federated_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -286,6 +327,8 @@ py_test(
         "//tensorflow_federated/python/core/impl/federated_context:federated_computation",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -302,7 +345,10 @@ py_library(
 py_test(
     name = "program_state_manager_test",
     srcs = ["program_state_manager_test.py"],
-    deps = [":program_state_manager"],
+    deps = [
+        ":program_state_manager",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 py_library(
@@ -313,6 +359,10 @@ py_library(
         ":value_reference",
         "//tensorflow_federated/python/common_libs:serializable",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -323,6 +373,8 @@ py_library(
         ":structure_utils",
         ":value_reference",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
     ],
 )
 
@@ -332,6 +384,10 @@ py_test(
     deps = [
         ":program_test_utils",
         ":release_manager",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -344,6 +400,7 @@ py_library(
         "//tensorflow_federated/python/common_libs:serializable",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:type_serialization",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -354,12 +411,19 @@ py_test(
         ":program_test_utils",
         ":serialization_utils",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "structure_utils",
     srcs = ["structure_utils.py"],
+    deps = [
+        "@pypi_attrs//:pkg",
+        "@pypi_dm_tree//:pkg",
+    ],
 )
 
 py_test(
@@ -368,6 +432,10 @@ py_test(
     deps = [
         ":program_test_utils",
         ":structure_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -379,6 +447,8 @@ py_library(
         ":structure_utils",
         ":value_reference",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -388,6 +458,10 @@ py_test(
     deps = [
         ":program_test_utils",
         ":tensorboard_release_manager",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -398,6 +472,7 @@ py_library(
         ":structure_utils",
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:typed_object",
+        "@pypi_numpy//:pkg",
     ],
 )
 
@@ -407,5 +482,9 @@ py_test(
     deps = [
         ":program_test_utils",
         ":value_reference",
+        "@pypi_absl_py//:pkg",
+        "@pypi_dm_tree//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/BUILD b/tensorflow_federated/python/simulation/BUILD
index b4a9f0649d..b728e3ab38 100644
--- a/tensorflow_federated/python/simulation/BUILD
+++ b/tensorflow_federated/python/simulation/BUILD
@@ -64,19 +64,27 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:iterative_process",
         "//tensorflow_federated/python/learning/templates:learning_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "sampling_utils",
     srcs = ["sampling_utils.py"],
+    deps = ["@pypi_numpy//:pkg"],
 )
 
 py_test(
     name = "sampling_utils_test",
     size = "small",
     srcs = ["sampling_utils_test.py"],
-    deps = [":sampling_utils"],
+    deps = [
+        ":sampling_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -88,6 +96,7 @@ py_library(
         "//tensorflow_federated/python/core/templates:iterative_process",
         "//tensorflow_federated/python/program:program_state_manager",
         "//tensorflow_federated/python/program:release_manager",
+        "@pypi_absl_py//:pkg",
     ],
 )
 
@@ -103,5 +112,7 @@ py_test(
         "//tensorflow_federated/python/core/impl/types:computation_types",
         "//tensorflow_federated/python/core/impl/types:placements",
         "//tensorflow_federated/python/core/templates:iterative_process",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/baselines/BUILD b/tensorflow_federated/python/simulation/baselines/BUILD
index 7da0f77293..51b082166a 100644
--- a/tensorflow_federated/python/simulation/baselines/BUILD
+++ b/tensorflow_federated/python/simulation/baselines/BUILD
@@ -37,6 +37,8 @@ py_library(
     deps = [
         ":task_data",
         "//tensorflow_federated/python/learning/models:variable",
+        "@pypi_attrs//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -48,23 +50,30 @@ py_test(
         ":task_data",
         "//tensorflow_federated/python/learning/models:keras_utils",
         "//tensorflow_federated/python/simulation/datasets:client_data",
+        "@pypi_attrs//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "client_spec",
     srcs = ["client_spec.py"],
+    deps = ["@pypi_attrs//:pkg"],
 )
 
 py_library(
     name = "keras_metrics",
     srcs = ["keras_metrics.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
     name = "keras_metrics_test",
     srcs = ["keras_metrics_test.py"],
-    deps = [":keras_metrics"],
+    deps = [
+        ":keras_metrics",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -73,6 +82,8 @@ py_library(
     deps = [
         "//tensorflow_federated/python/core/impl/computation:computation_base",
         "//tensorflow_federated/python/simulation/datasets:client_data",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -82,5 +93,7 @@ py_test(
     deps = [
         ":task_data",
         "//tensorflow_federated/python/simulation/datasets:client_data",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/baselines/cifar100/BUILD b/tensorflow_federated/python/simulation/baselines/cifar100/BUILD
index 3e39ebfdf8..385d6b9c97 100644
--- a/tensorflow_federated/python/simulation/baselines/cifar100/BUILD
+++ b/tensorflow_federated/python/simulation/baselines/cifar100/BUILD
@@ -25,7 +25,10 @@ py_library(
 py_library(
     name = "image_classification_preprocessing",
     srcs = ["image_classification_preprocessing.py"],
-    deps = ["//tensorflow_federated/python/simulation/baselines:client_spec"],
+    deps = [
+        "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -36,6 +39,8 @@ py_test(
         ":image_classification_preprocessing",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -52,6 +57,7 @@ py_library(
         "//tensorflow_federated/python/simulation/datasets:cifar100",
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/models:resnet_models",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -64,5 +70,7 @@ py_test(
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
         "//tensorflow_federated/python/simulation/models:resnet_models",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/baselines/emnist/BUILD b/tensorflow_federated/python/simulation/baselines/emnist/BUILD
index 3eccaf2097..fd6654f1ca 100644
--- a/tensorflow_federated/python/simulation/baselines/emnist/BUILD
+++ b/tensorflow_federated/python/simulation/baselines/emnist/BUILD
@@ -38,6 +38,7 @@ py_library(
         "//tensorflow_federated/python/simulation/baselines:task_data",
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/datasets:emnist",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -49,6 +50,8 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -65,6 +68,7 @@ py_library(
         "//tensorflow_federated/python/simulation/baselines:task_data",
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/datasets:emnist",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -77,24 +81,33 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "emnist_models",
     srcs = ["emnist_models.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
     name = "emnist_models_test",
     srcs = ["emnist_models_test.py"],
-    deps = [":emnist_models"],
+    deps = [
+        ":emnist_models",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "emnist_preprocessing",
     srcs = ["emnist_preprocessing.py"],
-    deps = ["//tensorflow_federated/python/simulation/baselines:client_spec"],
+    deps = [
+        "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -105,5 +118,7 @@ py_test(
         ":emnist_preprocessing",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/baselines/landmark/BUILD b/tensorflow_federated/python/simulation/baselines/landmark/BUILD
index ff80ec9d04..1b8741f383 100644
--- a/tensorflow_federated/python/simulation/baselines/landmark/BUILD
+++ b/tensorflow_federated/python/simulation/baselines/landmark/BUILD
@@ -37,6 +37,7 @@ py_library(
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/datasets:gldv2",
         "//tensorflow_federated/python/simulation/models:mobilenet_v2",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -48,13 +49,18 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "landmark_preprocessing",
     srcs = ["landmark_preprocessing.py"],
-    deps = ["//tensorflow_federated/python/simulation/baselines:client_spec"],
+    deps = [
+        "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -64,5 +70,7 @@ py_test(
         ":landmark_preprocessing",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/baselines/shakespeare/BUILD b/tensorflow_federated/python/simulation/baselines/shakespeare/BUILD
index a4d672c4c7..0bfef28ec8 100644
--- a/tensorflow_federated/python/simulation/baselines/shakespeare/BUILD
+++ b/tensorflow_federated/python/simulation/baselines/shakespeare/BUILD
@@ -25,18 +25,25 @@ py_library(
 py_library(
     name = "char_prediction_models",
     srcs = ["char_prediction_models.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
     name = "char_prediction_models_test",
     srcs = ["char_prediction_models_test.py"],
-    deps = [":char_prediction_models"],
+    deps = [
+        ":char_prediction_models",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "char_prediction_preprocessing",
     srcs = ["char_prediction_preprocessing.py"],
-    deps = ["//tensorflow_federated/python/simulation/baselines:client_spec"],
+    deps = [
+        "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -47,6 +54,8 @@ py_test(
         ":char_prediction_preprocessing",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -64,6 +73,7 @@ py_library(
         "//tensorflow_federated/python/simulation/baselines:task_data",
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/datasets:shakespeare",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -75,5 +85,7 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/baselines/stackoverflow/BUILD b/tensorflow_federated/python/simulation/baselines/stackoverflow/BUILD
index 02e9a767fa..1690e83b5a 100644
--- a/tensorflow_federated/python/simulation/baselines/stackoverflow/BUILD
+++ b/tensorflow_federated/python/simulation/baselines/stackoverflow/BUILD
@@ -37,6 +37,7 @@ py_library(
     deps = [
         ":constants",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -48,6 +49,8 @@ py_test(
         ":tag_prediction_preprocessing",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -64,6 +67,7 @@ py_library(
         "//tensorflow_federated/python/simulation/baselines:task_data",
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/datasets:stackoverflow",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -74,18 +78,25 @@ py_test(
         ":tag_prediction_tasks",
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "word_prediction_models",
     srcs = ["word_prediction_models.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
     name = "word_prediction_models_test",
     srcs = ["word_prediction_models_test.py"],
-    deps = [":word_prediction_models"],
+    deps = [
+        ":word_prediction_models",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -94,6 +105,8 @@ py_library(
     deps = [
         ":constants",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_attrs//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -105,6 +118,8 @@ py_test(
         ":word_prediction_preprocessing",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -123,6 +138,7 @@ py_library(
         "//tensorflow_federated/python/simulation/baselines:task_data",
         "//tensorflow_federated/python/simulation/datasets:client_data",
         "//tensorflow_federated/python/simulation/datasets:stackoverflow",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -134,5 +150,7 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/simulation/baselines:baseline_task",
         "//tensorflow_federated/python/simulation/baselines:client_spec",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
diff --git a/tensorflow_federated/python/simulation/datasets/BUILD b/tensorflow_federated/python/simulation/datasets/BUILD
index 138ca50751..16fda56d07 100644
--- a/tensorflow_federated/python/simulation/datasets/BUILD
+++ b/tensorflow_federated/python/simulation/datasets/BUILD
@@ -45,6 +45,7 @@ py_library(
     deps = [
         ":download",
         ":sql_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -52,7 +53,11 @@ py_test(
     name = "celeba_test",
     size = "small",
     srcs = ["celeba_test.py"],
-    deps = [":celeba"],
+    deps = [
+        ":celeba",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -62,6 +67,7 @@ py_library(
         ":download",
         ":from_tensor_slices_client_data",
         ":sql_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -69,12 +75,17 @@ py_test(
     name = "cifar100_test",
     size = "small",
     srcs = ["cifar100_test.py"],
-    deps = [":cifar100"],
+    deps = [
+        ":cifar100",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "dataset_utils",
     srcs = ["dataset_utils.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
@@ -84,19 +95,28 @@ py_test(
     deps = [
         ":dataset_utils",
         ":from_tensor_slices_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_library(
     name = "download",
     srcs = ["download.py"],
+    deps = [
+        "@pypi_tensorflow//:pkg",
+        "@pypi_tqdm//:pkg",
+    ],
 )
 
 py_test(
     name = "download_test",
     size = "small",
     srcs = ["download_test.py"],
-    deps = [":download"],
+    deps = [
+        ":download",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -106,6 +126,7 @@ py_library(
         ":download",
         ":from_tensor_slices_client_data",
         ":sql_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -113,7 +134,11 @@ py_test(
     name = "emnist_test",
     timeout = "long",
     srcs = ["emnist_test.py"],
-    deps = [":emnist"],
+    deps = [
+        ":emnist",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -122,13 +147,18 @@ py_library(
     deps = [
         ":client_data",
         ":file_per_user_client_data",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_test(
     name = "flair_test",
     srcs = ["flair_test.py"],
-    deps = [":flair"],
+    deps = [
+        ":flair",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -139,6 +169,7 @@ py_library(
         ":download",
         ":from_tensor_slices_client_data",
         ":sql_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -146,7 +177,11 @@ py_test(
     name = "shakespeare_test",
     size = "small",
     srcs = ["shakespeare_test.py"],
-    deps = [":shakespeare"],
+    deps = [
+        ":shakespeare",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -156,6 +191,7 @@ py_library(
         ":download",
         ":from_tensor_slices_client_data",
         ":sql_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -163,7 +199,11 @@ py_test(
     name = "stackoverflow_test",
     size = "medium",
     srcs = ["stackoverflow_test.py"],
-    deps = [":stackoverflow"],
+    deps = [
+        ":stackoverflow",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -173,6 +213,7 @@ py_library(
         ":client_data",
         ":from_tensor_slices_client_data",
         ":vision_datasets_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -183,6 +224,7 @@ py_test(
     deps = [
         ":gldv2",
         ":vision_datasets_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -193,6 +235,7 @@ py_library(
         ":client_data",
         ":from_tensor_slices_client_data",
         ":vision_datasets_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -203,6 +246,7 @@ py_test(
     deps = [
         ":inaturalist",
         ":vision_datasets_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -212,13 +256,17 @@ py_library(
     deps = [
         ":client_data",
         ":file_per_user_client_data",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_test(
     name = "vision_datasets_utils_test",
     srcs = ["vision_datasets_utils_test.py"],
-    deps = [":vision_datasets_utils"],
+    deps = [
+        ":vision_datasets_utils",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -228,6 +276,9 @@ py_library(
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
         "//tensorflow_federated/python/core/impl/computation:computation_base",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -239,6 +290,8 @@ py_test(
         ":client_data",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/environments/tensorflow_frontend:tensorflow_computation",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -249,6 +302,7 @@ py_library(
         ":client_data",
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "//tensorflow_federated/python/learning:tensor_utils",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -260,6 +314,7 @@ py_test(
         ":file_per_user_client_data",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/computation:computation_base",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -269,6 +324,7 @@ py_library(
     deps = [
         ":client_data",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -281,6 +337,9 @@ py_test(
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
         "//tensorflow_federated/python/core/impl/computation:computation_base",
         "//tensorflow_federated/python/core/impl/types:computation_types",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -290,6 +349,8 @@ py_library(
     deps = [
         ":client_data",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -300,6 +361,8 @@ py_test(
     deps = [
         ":sql_client_data",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -309,6 +372,7 @@ py_library(
     deps = [
         ":client_data",
         "//tensorflow_federated/python/common_libs:py_typecheck",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -319,6 +383,7 @@ py_test(
         ":from_tensor_slices_client_data",
         ":transforming_client_data",
         "//tensorflow_federated/python/core/backends/native:execution_contexts",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -328,11 +393,18 @@ py_library(
     deps = [
         ":client_data",
         ":sql_client_data",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_test(
     name = "sql_client_data_utils_test",
     srcs = ["sql_client_data_utils_test.py"],
-    deps = [":sql_client_data_utils"],
+    deps = [
+        ":sql_client_data_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
diff --git a/tensorflow_federated/python/simulation/models/BUILD b/tensorflow_federated/python/simulation/models/BUILD
index d62133aca9..f204d537e1 100644
--- a/tensorflow_federated/python/simulation/models/BUILD
+++ b/tensorflow_federated/python/simulation/models/BUILD
@@ -26,27 +26,42 @@ py_library(
 py_library(
     name = "mnist",
     srcs = ["mnist.py"],
-    deps = ["//tensorflow_federated/python/learning/metrics:counters"],
+    deps = [
+        "//tensorflow_federated/python/learning/metrics:counters",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "mobilenet_v2",
     srcs = ["mobilenet_v2.py"],
+    deps = [
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "mobilenet_v2_test",
     srcs = ["mobilenet_v2_test.py"],
-    deps = [":mobilenet_v2"],
+    deps = [
+        ":mobilenet_v2",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "resnet_models",
     srcs = ["resnet_models.py"],
+    deps = [
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "resnet_models_test",
     srcs = ["resnet_models_test.py"],
-    deps = [":resnet_models"],
+    deps = [
+        ":resnet_models",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
diff --git a/tensorflow_federated/python/tensorflow_libs/BUILD b/tensorflow_federated/python/tensorflow_libs/BUILD
index 25f814b09b..67689e3108 100644
--- a/tensorflow_federated/python/tensorflow_libs/BUILD
+++ b/tensorflow_federated/python/tensorflow_libs/BUILD
@@ -24,12 +24,17 @@ py_library(
 py_library(
     name = "graph_utils",
     srcs = ["graph_utils.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
 
 py_test(
     name = "graph_utils_test",
     srcs = ["graph_utils_test.py"],
-    deps = [":graph_utils"],
+    deps = [
+        ":graph_utils",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
@@ -38,6 +43,7 @@ py_library(
     deps = [
         "//tensorflow_federated/python/common_libs:py_typecheck",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -48,6 +54,8 @@ py_test(
     deps = [
         ":serialization_utils",
         "@com_google_protobuf//:protobuf_python",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
@@ -55,4 +63,5 @@ py_library(
     name = "tensorflow_test_utils",
     testonly = True,
     srcs = ["tensorflow_test_utils.py"],
+    deps = ["@pypi_tensorflow//:pkg"],
 )
diff --git a/tensorflow_federated/python/tests/BUILD b/tensorflow_federated/python/tests/BUILD
index 5fbccb468b..7f530e9201 100644
--- a/tensorflow_federated/python/tests/BUILD
+++ b/tensorflow_federated/python/tests/BUILD
@@ -17,13 +17,21 @@ py_test(
     name = "ast_generation_test",
     size = "small",
     srcs = ["ast_generation_test.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "async_execution_context_integration_test",
     srcs = ["async_execution_context_integration_test.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+    ],
 )
 
 py_test(
@@ -34,20 +42,32 @@ py_test(
         ":temperature_sensor_example",
         ":test_contexts",
         "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
     ],
 )
 
 py_cpu_gpu_test(
     name = "backend_accelerators_test",
     srcs = ["backend_accelerators_test.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "map_reduce_form_test",
     size = "medium",
     srcs = ["map_reduce_form_test.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -55,7 +75,11 @@ py_test(
     size = "small",
     timeout = "long",
     srcs = ["mergeable_comp_execution_context_integration_test.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+    ],
 )
 
 py_test(
@@ -64,7 +88,11 @@ py_test(
     timeout = "long",
     srcs = ["simulation_test.py"],
     shard_count = 4,
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
@@ -73,20 +101,34 @@ py_test(
     timeout = "moderate",
     srcs = ["remote_runtime_stream_structs_test.py"],
     tags = ["requires-mem:20g"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_test(
     name = "sync_local_cpp_execution_context_test",
     size = "small",
     srcs = ["sync_local_cpp_execution_context_test.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_absl_py//:pkg",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
     name = "temperature_sensor_example",
     srcs = ["temperature_sensor_example.py"],
-    deps = ["//tensorflow_federated"],
+    deps = [
+        "//tensorflow_federated",
+        "@pypi_numpy//:pkg",
+        "@pypi_tensorflow//:pkg",
+    ],
 )
 
 py_library(
diff --git a/third_party/grpc/BUILD b/third_party/grpc/BUILD
new file mode 100644
index 0000000000..31d4c71147
--- /dev/null
+++ b/third_party/grpc/BUILD
@@ -0,0 +1,6 @@
+package(
+    default_applicable_licenses = ["//:package_license"],
+    default_visibility = ["//visibility:private"],
+)
+
+licenses(["notice"])
diff --git a/third_party/grpc/grpc_noexcept.patch b/third_party/grpc/grpc_noexcept.patch
new file mode 100644
index 0000000000..d82fd6c460
--- /dev/null
+++ b/third_party/grpc/grpc_noexcept.patch
@@ -0,0 +1,462 @@
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pxd.pxi src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pxd.pxi
+index e5620cd166..7bb2bd71de 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pxd.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pxd.pxi
+@@ -48,7 +48,7 @@ cdef class CallbackWrapper:
+     @staticmethod
+     cdef void functor_run(
+             grpc_experimental_completion_queue_functor* functor,
+-            int succeed)
++            int succeed) noexcept
+ 
+     cdef grpc_experimental_completion_queue_functor *c_functor(self)
+ 
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pyx.pxi src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pyx.pxi
+index 86fc91e76a..475bc5520d 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pyx.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/aio/callback_common.pyx.pxi
+@@ -50,7 +50,7 @@ cdef class CallbackWrapper:
+     @staticmethod
+     cdef void functor_run(
+             grpc_experimental_completion_queue_functor* functor,
+-            int success):
++            int success) noexcept:
+         cdef CallbackContext *context = <CallbackContext *>functor
+         cdef object waiter = <object>context.waiter
+         if not waiter.cancelled():
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/aio/iomgr/iomgr.pyx.pxi src/python/grpcio/grpc/_cython/_cygrpc/aio/iomgr/iomgr.pyx.pxi
+index f5b62af528..d84b7e0ab0 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/aio/iomgr/iomgr.pyx.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/aio/iomgr/iomgr.pyx.pxi
+@@ -33,14 +33,14 @@ cdef bint so_reuse_port
+ 
+ cdef grpc_error* asyncio_socket_init(
+         grpc_custom_socket* grpc_socket,
+-        int domain) with gil:
++        int domain) noexcept with gil:
+     socket = _AsyncioSocket.create(grpc_socket, None, None)
+     Py_INCREF(socket)
+     grpc_socket.impl = <void*>socket
+     return <grpc_error*>0
+ 
+ 
+-cdef void asyncio_socket_destroy(grpc_custom_socket* grpc_socket) with gil:
++cdef void asyncio_socket_destroy(grpc_custom_socket* grpc_socket) noexcept with gil:
+     Py_DECREF(<_AsyncioSocket>grpc_socket.impl)
+ 
+ 
+@@ -48,7 +48,7 @@ cdef void asyncio_socket_connect(
+         grpc_custom_socket* grpc_socket,
+         const grpc_sockaddr* addr,
+         size_t addr_len,
+-        grpc_custom_connect_callback connect_cb) with gil:
++        grpc_custom_connect_callback connect_cb) noexcept with gil:
+     host, port = sockaddr_to_tuple(addr, addr_len)
+     socket = <_AsyncioSocket>grpc_socket.impl
+     socket.connect(host, port, connect_cb)
+@@ -56,13 +56,13 @@ cdef void asyncio_socket_connect(
+ 
+ cdef void asyncio_socket_close(
+         grpc_custom_socket* grpc_socket,
+-        grpc_custom_close_callback close_cb) with gil:
++        grpc_custom_close_callback close_cb) noexcept with gil:
+     socket = (<_AsyncioSocket>grpc_socket.impl)
+     socket.close()
+     close_cb(grpc_socket)
+ 
+ 
+-cdef void asyncio_socket_shutdown(grpc_custom_socket* grpc_socket) with gil:
++cdef void asyncio_socket_shutdown(grpc_custom_socket* grpc_socket) noexcept with gil:
+     socket = (<_AsyncioSocket>grpc_socket.impl)
+     socket.close()
+ 
+@@ -70,7 +70,7 @@ cdef void asyncio_socket_shutdown(grpc_custom_socket* grpc_socket) with gil:
+ cdef void asyncio_socket_write(
+         grpc_custom_socket* grpc_socket,
+         grpc_slice_buffer* slice_buffer,
+-        grpc_custom_write_callback write_cb) with gil:
++        grpc_custom_write_callback write_cb) noexcept with gil:
+     socket = (<_AsyncioSocket>grpc_socket.impl)
+     socket.write(slice_buffer, write_cb)
+ 
+@@ -79,7 +79,7 @@ cdef void asyncio_socket_read(
+         grpc_custom_socket* grpc_socket,
+         char* buffer_,
+         size_t length,
+-        grpc_custom_read_callback read_cb) with gil:
++        grpc_custom_read_callback read_cb) noexcept with gil:
+     socket = (<_AsyncioSocket>grpc_socket.impl)
+     socket.read(buffer_, length, read_cb)
+ 
+@@ -87,7 +87,7 @@ cdef void asyncio_socket_read(
+ cdef grpc_error* asyncio_socket_getpeername(
+         grpc_custom_socket* grpc_socket,
+         const grpc_sockaddr* addr,
+-        int* length) with gil:
++        int* length) noexcept with gil:
+     peer = (<_AsyncioSocket>grpc_socket.impl).peername()
+ 
+     cdef grpc_resolved_address c_addr
+@@ -102,7 +102,7 @@ cdef grpc_error* asyncio_socket_getpeername(
+ cdef grpc_error* asyncio_socket_getsockname(
+         grpc_custom_socket* grpc_socket,
+         const grpc_sockaddr* addr,
+-        int* length) with gil:
++        int* length) noexcept with gil:
+     """Supplies sock_addr in add_socket_to_server."""
+     cdef grpc_resolved_address c_addr
+     socket = (<_AsyncioSocket>grpc_socket.impl)
+@@ -118,7 +118,7 @@ cdef grpc_error* asyncio_socket_getsockname(
+     return grpc_error_none()
+ 
+ 
+-cdef grpc_error* asyncio_socket_listen(grpc_custom_socket* grpc_socket) with gil:
++cdef grpc_error* asyncio_socket_listen(grpc_custom_socket* grpc_socket) noexcept with gil:
+     (<_AsyncioSocket>grpc_socket.impl).listen()
+     return grpc_error_none()
+ 
+@@ -137,7 +137,7 @@ def _asyncio_apply_socket_options(object s, int flags):
+ cdef grpc_error* asyncio_socket_bind(
+         grpc_custom_socket* grpc_socket,
+         const grpc_sockaddr* addr,
+-        size_t len, int flags) with gil:
++        size_t len, int flags) noexcept with gil:
+     host, port = sockaddr_to_tuple(addr, len)
+     try:
+         ip = ipaddress.ip_address(host)
+@@ -162,14 +162,14 @@ cdef grpc_error* asyncio_socket_bind(
+ cdef void asyncio_socket_accept(
+         grpc_custom_socket* grpc_socket,
+         grpc_custom_socket* grpc_socket_client,
+-        grpc_custom_accept_callback accept_cb) with gil:
++        grpc_custom_accept_callback accept_cb) noexcept with gil:
+     (<_AsyncioSocket>grpc_socket.impl).accept(grpc_socket_client, accept_cb)
+ 
+ 
+ cdef grpc_error* asyncio_resolve(
+         char* host,
+         char* port,
+-        grpc_resolved_addresses** res) with gil:
++        grpc_resolved_addresses** res) noexcept with gil:
+     result = native_socket.getaddrinfo(host, port)
+     res[0] = tuples_to_resolvaddr(result)
+ 
+@@ -177,17 +177,17 @@ cdef grpc_error* asyncio_resolve(
+ cdef void asyncio_resolve_async(
+         grpc_custom_resolver* grpc_resolver,
+         char* host,
+-        char* port) with gil:
++        char* port) noexcept with gil:
+     resolver = _AsyncioResolver.create(grpc_resolver)
+     resolver.resolve(host, port)
+ 
+ 
+-cdef void asyncio_timer_start(grpc_custom_timer* grpc_timer) with gil:
++cdef void asyncio_timer_start(grpc_custom_timer* grpc_timer) noexcept with gil:
+     timer = _AsyncioTimer.create(grpc_timer, grpc_timer.timeout_ms / 1000.0)
+     grpc_timer.timer = <void*>timer
+ 
+ 
+-cdef void asyncio_timer_stop(grpc_custom_timer* grpc_timer) with gil:
++cdef void asyncio_timer_stop(grpc_custom_timer* grpc_timer) noexcept with gil:
+     # TODO(https://github.com/grpc/grpc/issues/22278) remove this if condition
+     if grpc_timer.timer == NULL:
+         return
+@@ -196,19 +196,19 @@ cdef void asyncio_timer_stop(grpc_custom_timer* grpc_timer) with gil:
+         timer.stop()
+ 
+ 
+-cdef void asyncio_init_loop() with gil:
++cdef void asyncio_init_loop() noexcept with gil:
+     pass
+ 
+ 
+-cdef void asyncio_destroy_loop() with gil:
++cdef void asyncio_destroy_loop() noexcept with gil:
+     pass
+ 
+ 
+-cdef void asyncio_kick_loop() with gil:
++cdef void asyncio_kick_loop() noexcept with gil:
+     pass
+ 
+ 
+-cdef void asyncio_run_loop(size_t timeout_ms) with gil:
++cdef void asyncio_run_loop(size_t timeout_ms) noexcept with gil:
+     pass
+ 
+ 
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/credentials.pyx.pxi src/python/grpcio/grpc/_cython/_cygrpc/credentials.pyx.pxi
+index 24d1e2a3b7..641568b75a 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/credentials.pyx.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/credentials.pyx.pxi
+@@ -293,7 +293,7 @@ def server_credentials_ssl_dynamic_cert_config(initial_cert_config,
+   return credentials
+ 
+ cdef grpc_ssl_certificate_config_reload_status _server_cert_config_fetcher_wrapper(
+-        void* user_data, grpc_ssl_server_certificate_config **config) with gil:
++        void* user_data, grpc_ssl_server_certificate_config **config) noexcept with gil:
+   # This is a credentials.ServerCertificateConfig
+   cdef ServerCertificateConfig cert_config = None
+   if not user_data:
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pxd.pxi src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pxd.pxi
+index a925bdd2e6..09ff7d4604 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pxd.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pxd.pxi
+@@ -15,15 +15,15 @@
+ 
+ cdef extern from "pthread.h" nogil:
+     int pthread_atfork(
+-        void (*prepare)() nogil,
+-        void (*parent)() nogil,
+-        void (*child)() nogil)
++        void (*prepare)() noexcept nogil,
++        void (*parent)() noexcept nogil,
++        void (*child)() noexcept nogil)
+ 
+ 
+-cdef void __prefork() nogil
++cdef void __prefork() noexcept nogil
+ 
+ 
+-cdef void __postfork_parent() nogil
++cdef void __postfork_parent() noexcept nogil
+ 
+ 
+-cdef void __postfork_child() nogil
++cdef void __postfork_child() noexcept nogil
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pyx.pxi src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pyx.pxi
+index 55c8673dd4..0ef3a5fdb1 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pyx.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/fork_posix.pyx.pxi
+@@ -34,7 +34,7 @@ _GRPC_ENABLE_FORK_SUPPORT = (
+ 
+ _fork_handler_failed = False
+ 
+-cdef void __prefork() nogil:
++cdef void __prefork() noexcept nogil:
+     with gil:
+         global _fork_handler_failed
+         _fork_handler_failed = False
+@@ -48,14 +48,14 @@ cdef void __prefork() nogil:
+             _fork_handler_failed = True
+ 
+ 
+-cdef void __postfork_parent() nogil:
++cdef void __postfork_parent() noexcept nogil:
+     with gil:
+         with _fork_state.fork_in_progress_condition:
+             _fork_state.fork_in_progress = False
+             _fork_state.fork_in_progress_condition.notify_all()
+ 
+ 
+-cdef void __postfork_child() nogil:
++cdef void __postfork_child() noexcept nogil:
+     with gil:
+         try:
+             if _fork_handler_failed:
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/grpc_gevent.pyx.pxi src/python/grpcio/grpc/_cython/_cygrpc/grpc_gevent.pyx.pxi
+index 05e185ddd4..7b9521def7 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/grpc_gevent.pyx.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/grpc_gevent.pyx.pxi
+@@ -41,7 +41,7 @@ cdef class SocketWrapper:
+   def __dealloc__(self):
+     grpc_shutdown_blocking()
+ 
+-cdef grpc_error* socket_init(grpc_custom_socket* socket, int domain) with gil:
++cdef grpc_error* socket_init(grpc_custom_socket* socket, int domain) noexcept with gil:
+   sw = SocketWrapper()
+   sw.c_socket = socket
+   sw.sockopts = []
+@@ -69,7 +69,7 @@ def socket_connect_async(socket_wrapper, addr_tuple):
+ 
+ cdef void socket_connect(grpc_custom_socket* socket, const grpc_sockaddr* addr,
+                          size_t addr_len,
+-                         grpc_custom_connect_callback cb) with gil:
++                         grpc_custom_connect_callback cb) noexcept with gil:
+   py_socket = None
+   socket_wrapper = <SocketWrapper>socket.impl
+   socket_wrapper.connect_cb = cb
+@@ -82,10 +82,10 @@ cdef void socket_connect(grpc_custom_socket* socket, const grpc_sockaddr* addr,
+   socket_wrapper.socket = py_socket
+   _spawn_greenlet(socket_connect_async, socket_wrapper, addr_tuple)
+ 
+-cdef void socket_destroy(grpc_custom_socket* socket) with gil:
++cdef void socket_destroy(grpc_custom_socket* socket) noexcept with gil:
+   cpython.Py_DECREF(<SocketWrapper>socket.impl)
+ 
+-cdef void socket_shutdown(grpc_custom_socket* socket) with gil:
++cdef void socket_shutdown(grpc_custom_socket* socket) noexcept with gil:
+   try:
+     (<SocketWrapper>socket.impl).socket.shutdown(gevent_socket.SHUT_RDWR)
+   except IOError as io_error:
+@@ -93,7 +93,7 @@ cdef void socket_shutdown(grpc_custom_socket* socket) with gil:
+       raise io_error
+ 
+ cdef void socket_close(grpc_custom_socket* socket,
+-                       grpc_custom_close_callback cb) with gil:
++                       grpc_custom_close_callback cb) noexcept with gil:
+   socket_wrapper = (<SocketWrapper>socket.impl)
+   if socket_wrapper.socket is not None:
+     socket_wrapper.socket.close()
+@@ -133,7 +133,7 @@ def socket_write_async(socket_wrapper, write_bytes):
+   socket_write_async_cython(socket_wrapper, write_bytes)
+ 
+ cdef void socket_write(grpc_custom_socket* socket, grpc_slice_buffer* buffer,
+-                       grpc_custom_write_callback cb) with gil:
++                       grpc_custom_write_callback cb) noexcept with gil:
+   cdef char* start
+   sw = <SocketWrapper>socket.impl
+   sw.write_cb = cb
+@@ -161,7 +161,7 @@ def socket_read_async(socket_wrapper):
+   socket_read_async_cython(socket_wrapper)
+ 
+ cdef void socket_read(grpc_custom_socket* socket, char* buffer,
+-                      size_t length, grpc_custom_read_callback cb) with gil:
++                      size_t length, grpc_custom_read_callback cb) noexcept with gil:
+   sw = <SocketWrapper>socket.impl
+   sw.read_cb = cb
+   sw.c_buffer = buffer
+@@ -170,7 +170,7 @@ cdef void socket_read(grpc_custom_socket* socket, char* buffer,
+ 
+ cdef grpc_error* socket_getpeername(grpc_custom_socket* socket,
+                                     const grpc_sockaddr* addr,
+-                                    int* length) with gil:
++                                    int* length) noexcept with gil:
+   cdef char* src_buf
+   peer = (<SocketWrapper>socket.impl).socket.getpeername()
+ 
+@@ -183,7 +183,7 @@ cdef grpc_error* socket_getpeername(grpc_custom_socket* socket,
+ 
+ cdef grpc_error* socket_getsockname(grpc_custom_socket* socket,
+                                     const grpc_sockaddr* addr,
+-                                    int* length) with gil:
++                                    int* length) noexcept with gil:
+   cdef char* src_buf
+   cdef grpc_resolved_address c_addr
+   if (<SocketWrapper>socket.impl).socket is None:
+@@ -202,7 +202,7 @@ def applysockopts(s):
+ 
+ cdef grpc_error* socket_bind(grpc_custom_socket* socket,
+                              const grpc_sockaddr* addr,
+-                             size_t len, int flags) with gil:
++                             size_t len, int flags) noexcept with gil:
+   addr_tuple = sockaddr_to_tuple(addr, len)
+   try:
+     try:
+@@ -219,7 +219,7 @@ cdef grpc_error* socket_bind(grpc_custom_socket* socket,
+   else:
+     return grpc_error_none()
+ 
+-cdef grpc_error* socket_listen(grpc_custom_socket* socket) with gil:
++cdef grpc_error* socket_listen(grpc_custom_socket* socket) noexcept with gil:
+   (<SocketWrapper>socket.impl).socket.listen(50)
+   return grpc_error_none()
+ 
+@@ -249,7 +249,7 @@ def socket_accept_async(s):
+   accept_callback_cython(s)
+ 
+ cdef void socket_accept(grpc_custom_socket* socket, grpc_custom_socket* client,
+-                        grpc_custom_accept_callback cb) with gil:
++                        grpc_custom_accept_callback cb) noexcept with gil:
+   sw = <SocketWrapper>socket.impl
+   sw.accepting_socket = client
+   sw.accept_cb = cb
+@@ -283,7 +283,7 @@ cdef socket_resolve_async_cython(ResolveWrapper resolve_wrapper):
+ def socket_resolve_async_python(resolve_wrapper):
+   socket_resolve_async_cython(resolve_wrapper)
+ 
+-cdef void socket_resolve_async(grpc_custom_resolver* r, char* host, char* port) with gil:
++cdef void socket_resolve_async(grpc_custom_resolver* r, char* host, char* port) noexcept with gil:
+   rw = ResolveWrapper()
+   rw.c_resolver = r
+   rw.c_host = host
+@@ -291,7 +291,7 @@ cdef void socket_resolve_async(grpc_custom_resolver* r, char* host, char* port)
+   _spawn_greenlet(socket_resolve_async_python, rw)
+ 
+ cdef grpc_error* socket_resolve(char* host, char* port,
+-                                grpc_resolved_addresses** res) with gil:
++                                grpc_resolved_addresses** res) noexcept with gil:
+     try:
+       result = gevent_socket.getaddrinfo(host, port)
+       res[0] = tuples_to_resolvaddr(result)
+@@ -325,13 +325,13 @@ cdef class TimerWrapper:
+   def __dealloc__(self):
+     grpc_shutdown_blocking()
+ 
+-cdef void timer_start(grpc_custom_timer* t) with gil:
++cdef void timer_start(grpc_custom_timer* t) noexcept with gil:
+   timer = TimerWrapper(t.timeout_ms / 1000.0)
+   timer.c_timer = t
+   t.timer = <void*>timer
+   timer.start()
+ 
+-cdef void timer_stop(grpc_custom_timer* t) with gil:
++cdef void timer_stop(grpc_custom_timer* t) noexcept with gil:
+   time_wrapper = <object>t.timer
+   time_wrapper.stop()
+ 
+@@ -339,16 +339,16 @@ cdef void timer_stop(grpc_custom_timer* t) with gil:
+ ### pollset implementation ###
+ ###############################
+ 
+-cdef void init_loop() with gil:
++cdef void init_loop() noexcept with gil:
+   pass
+ 
+-cdef void destroy_loop() with gil:
++cdef void destroy_loop() noexcept with gil:
+   g_pool.join()
+ 
+-cdef void kick_loop() with gil:
++cdef void kick_loop() noexcept with gil:
+   g_event.set()
+ 
+-cdef void run_loop(size_t timeout_ms) with gil:
++cdef void run_loop(size_t timeout_ms) noexcept with gil:
+     timeout = timeout_ms / 1000.0
+     if timeout_ms > 0:
+       g_event.wait(timeout)
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/vtable.pxd.pxi src/python/grpcio/grpc/_cython/_cygrpc/vtable.pxd.pxi
+index c96e5cb669..3dfd7e5ff1 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/vtable.pxd.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/vtable.pxd.pxi
+@@ -13,11 +13,11 @@
+ # limitations under the License.
+ 
+ 
+-cdef void* _copy_pointer(void* pointer)
++cdef void* _copy_pointer(void* pointer) noexcept
+ 
+-cdef void _destroy_pointer(void* pointer)
++cdef void _destroy_pointer(void* pointer) noexcept
+ 
+-cdef int _compare_pointer(void* first_pointer, void* second_pointer)
++cdef int _compare_pointer(void* first_pointer, void* second_pointer) noexcept
+ 
+ 
+ cdef grpc_arg_pointer_vtable default_vtable
+diff --git src/python/grpcio/grpc/_cython/_cygrpc/vtable.pyx.pxi src/python/grpcio/grpc/_cython/_cygrpc/vtable.pyx.pxi
+index da4b81bd97..f59410073b 100644
+--- src/python/grpcio/grpc/_cython/_cygrpc/vtable.pyx.pxi
++++ src/python/grpcio/grpc/_cython/_cygrpc/vtable.pyx.pxi
+@@ -13,16 +13,16 @@
+ # limitations under the License.
+ 
+ # TODO(https://github.com/grpc/grpc/issues/15662): Reform this.
+-cdef void* _copy_pointer(void* pointer):
++cdef void* _copy_pointer(void* pointer) noexcept:
+   return pointer
+ 
+ 
+ # TODO(https://github.com/grpc/grpc/issues/15662): Reform this.
+-cdef void _destroy_pointer(void* pointer):
++cdef void _destroy_pointer(void* pointer) noexcept:
+   pass
+ 
+ 
+-cdef int _compare_pointer(void* first_pointer, void* second_pointer):
++cdef int _compare_pointer(void* first_pointer, void* second_pointer) noexcept:
+   if first_pointer < second_pointer:
+     return -1
+   elif first_pointer > second_pointer:
diff --git a/third_party/grpc/register_go_toolchain.patch b/third_party/grpc/register_go_toolchain.patch
new file mode 100644
index 0000000000..e12008fbae
--- /dev/null
+++ b/third_party/grpc/register_go_toolchain.patch
@@ -0,0 +1,28 @@
+diff --git bazel/grpc_extra_deps.bzl bazel/grpc_extra_deps.bzl
+index 4c1dfad2e8..40bc00c184 100644
+--- bazel/grpc_extra_deps.bzl
++++ bazel/grpc_extra_deps.bzl
+@@ -1,11 +1,11 @@
+ """Loads the dependencies necessary for the external repositories defined in grpc_deps.bzl."""
+
++load("@build_bazel_apple_support//lib:repositories.bzl", "apple_support_dependencies")
++load("@build_bazel_rules_apple//apple:repositories.bzl", "apple_rules_dependencies")
+ load("@com_google_protobuf//:protobuf_deps.bzl", "protobuf_deps")
+-load("@upb//bazel:workspace_deps.bzl", "upb_deps")
+ load("@envoy_api//bazel:repositories.bzl", "api_dependencies")
+ load("@io_bazel_rules_go//go:deps.bzl", "go_register_toolchains", "go_rules_dependencies")
+-load("@build_bazel_rules_apple//apple:repositories.bzl", "apple_rules_dependencies")
+-load("@build_bazel_apple_support//lib:repositories.bzl", "apple_support_dependencies")
++load("@upb//bazel:workspace_deps.bzl", "upb_deps")
+
+ def grpc_extra_deps():
+     """Loads the extra dependencies.
+@@ -33,7 +33,7 @@ def grpc_extra_deps():
+     api_dependencies()
+
+     go_rules_dependencies()
+-    go_register_toolchains()
++    go_register_toolchains(version = "1.18.4")
+
+     apple_rules_dependencies()
+
diff --git a/third_party/tensorflow/internal_visibility.patch b/third_party/tensorflow/internal_visibility.patch
index 7703001ef6..a1cea88be3 100644
--- a/third_party/tensorflow/internal_visibility.patch
+++ b/third_party/tensorflow/internal_visibility.patch
@@ -1,13 +1,13 @@
 diff --git tensorflow/BUILD tensorflow/BUILD
-index 202553cd531..171eb04665c 100644
+index 5018c6e642e..cf3a1ca0760 100644
 --- tensorflow/BUILD
 +++ tensorflow/BUILD
-@@ -1039,7 +1039,7 @@ package_group(
+@@ -1057,7 +1057,7 @@ package_group(
          "//smartass/brain/configure/...",
          "//tensorflow/...",
          "//tensorflow_decision_forests/...",
 -        "//tensorflow_federated/...",
 +        "public",
+         "//tensorflow_models/google/recml/...",
          "//third_party/cloud_tpu/convergence_tools/sdc_monitoring/...",
          "//third_party/cloud_tpu/inference_converter/...",
-         "//third_party/py/cloud_ml_autoflow/...",
diff --git a/third_party/tensorflow/python_toolchain.patch b/third_party/tensorflow/python_toolchain.patch
index 4d732d447d..a2d60d1b4a 100644
--- a/third_party/tensorflow/python_toolchain.patch
+++ b/third_party/tensorflow/python_toolchain.patch
@@ -1,5 +1,5 @@
 diff --git tensorflow/tools/toolchains/cpus/aarch64/aarch64_compiler_configure.bzl tensorflow/tools/toolchains/cpus/aarch64/aarch64_compiler_configure.bzl
-index a2bdd6a7eed..ec25c23d8d4 100644
+index 9261a652f9c..0557e345ab1 100644
 --- tensorflow/tools/toolchains/cpus/aarch64/aarch64_compiler_configure.bzl
 +++ tensorflow/tools/toolchains/cpus/aarch64/aarch64_compiler_configure.bzl
 @@ -2,7 +2,7 @@
@@ -24,23 +24,23 @@ index 9f71a414bf7..57f70752323 100644
  load("//third_party/nccl:nccl_configure.bzl", "remote_nccl_configure")
  load("//third_party/gpus:rocm_configure.bzl", "remote_rocm_configure")
 diff --git tensorflow/workspace2.bzl tensorflow/workspace2.bzl
-index 7e9faa558a4..5b18cb0969a 100644
+index 9e15c75c183..facc1de3d73 100644
 --- tensorflow/workspace2.bzl
 +++ tensorflow/workspace2.bzl
-@@ -8,7 +8,7 @@ load("//third_party/gpus:rocm_configure.bzl", "rocm_configure")
- load("//third_party/tensorrt:tensorrt_configure.bzl", "tensorrt_configure")
+@@ -37,7 +37,7 @@ load("//third_party/nasm:workspace.bzl", nasm = "repo")
  load("//third_party/nccl:nccl_configure.bzl", "nccl_configure")
- load("//third_party/git:git_configure.bzl", "git_configure")
+ load("//third_party/opencl_headers:workspace.bzl", opencl_headers = "repo")
+ load("//third_party/pasta:workspace.bzl", pasta = "repo")
 -load("//third_party/py:python_configure.bzl", "python_configure")
 +load("//third_party/py/non_hermetic:python_configure.bzl", "python_configure")
- load("//third_party/systemlibs:syslibs_configure.bzl", "syslibs_configure")
- load("//tensorflow/tools/toolchains:cpus/aarch64/aarch64_compiler_configure.bzl", "aarch64_compiler_configure")
- load("//tensorflow/tools/toolchains:cpus/arm/arm_compiler_configure.bzl", "arm_compiler_configure")
+ load("//third_party/py/ml_dtypes:workspace.bzl", ml_dtypes = "repo")
+ load("//third_party/pybind11_abseil:workspace.bzl", pybind11_abseil = "repo")
+ load("//third_party/pybind11_bazel:workspace.bzl", pybind11_bazel = "repo")
 diff --git third_party/py/non_hermetic/python_configure.bzl third_party/py/non_hermetic/python_configure.bzl
-index 300cbfb6c71..09d98505dd9 100644
+index 89732c3e33d..4ac1c8f5c04 100644
 --- third_party/py/non_hermetic/python_configure.bzl
 +++ third_party/py/non_hermetic/python_configure.bzl
-@@ -206,7 +206,7 @@ def _create_local_python_repository(repository_ctx):
+@@ -203,7 +203,7 @@ def _create_local_python_repository(repository_ctx):
      # Resolve all labels before doing any real work. Resolving causes the
      # function to be restarted with all previous state being lost. This
      # can easily lead to a O(n^2) runtime in the number of labels.
@@ -72,3 +72,17 @@ index 97c7907fc38..c80cc5287bc 100644
  )
  
  genrule(
+diff --git third_party/py/python_configure.bzl third_party/py/python_configure.bzl
+index 3728a91b931..ecb0ecb60e8 100644
+--- third_party/py/python_configure.bzl
++++ third_party/py/python_configure.bzl
+@@ -14,7 +14,7 @@ def _create_local_python_repository(repository_ctx):
+     # Resolve all labels before doing any real work. Resolving causes the
+     # function to be restarted with all previous state being lost. This
+     # can easily lead to a O(n^2) runtime in the number of labels.
+-    build_tpl = repository_ctx.path(Label("//third_party/py:BUILD.tpl"))
++    build_tpl = repository_ctx.path(Label("//third_party/py/non_hermetic:BUILD.tpl"))
+     platform_constraint = ""
+     if repository_ctx.attr.platform_constraint:
+         platform_constraint = "\"%s\"" % repository_ctx.attr.platform_constraint
+(END)
diff --git a/third_party/tensorflow/tf2xla_visibility.patch b/third_party/tensorflow/tf2xla_visibility.patch
index 84b7cdcab3..51f6fdf219 100644
--- a/third_party/tensorflow/tf2xla_visibility.patch
+++ b/third_party/tensorflow/tf2xla_visibility.patch
@@ -1,8 +1,8 @@
 diff --git tensorflow/compiler/tf2xla/BUILD tensorflow/compiler/tf2xla/BUILD
-index 22d9877bed9..3f6b421465d 100644
+index c1a2243de20..f4e24260aea 100644
 --- tensorflow/compiler/tf2xla/BUILD
 +++ tensorflow/compiler/tf2xla/BUILD
-@@ -46,7 +46,7 @@ package_group(
+@@ -47,7 +47,7 @@ package_group(
      packages = [
          "//platforms/performance/automl/...",
          "//tensorflow/...",
diff --git a/tools/BUILD b/tools/BUILD
index c521e31621..4596aa5c55 100644
--- a/tools/BUILD
+++ b/tools/BUILD
@@ -25,6 +25,10 @@ bzl_library(
 py_binary(
     name = "build_docs",
     srcs = ["build_docs.py"],
+    deps = [
+        "//third_party/py/tensorflow_docs/api_generator",
+        "@pypi_absl_py//:pkg",
+    ],
 )
 
 filegroup(
diff --git a/tools/build_defs.bzl b/tools/build_defs.bzl
index e86ff89a8a..4ddf47afc0 100644
--- a/tools/build_defs.bzl
+++ b/tools/build_defs.bzl
@@ -14,6 +14,7 @@
 """TensorFlow Federated build macros and rules."""
 
 load("@rules_python//python:defs.bzl", "py_test")
+load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_copts")
 
 def if_static(extra, framework_shared_object = []):  # buildifier: disable=unused-variable
     return_value = {
@@ -68,12 +69,14 @@ def tff_cc_cpu_gpu_test(name, tags = [], **kwargs):
     """
     native.cc_test(
         name = name + "_cpu",
+        copts = tf_copts(),
         tags = tags,
         **kwargs
     )
     native.cc_test(
         name = name + "_gpu",
         tags = tags + ["requires-gpu-nvidia"],
+        copts = tf_copts(),
         **kwargs
     )
     native.test_suite(
@@ -83,3 +86,54 @@ def tff_cc_cpu_gpu_test(name, tags = [], **kwargs):
             name + "_gpu",
         ],
     )
+
+def tff_cc_library(name, tags = [], **kwargs):
+    """A version of `cc_library` that adds default copts.
+
+    It accepts all `cc_library` arguments.
+
+    Args:
+      name: A unique name for this target.
+      tags: List of arbitrary text tags.
+      **kwargs: `cc_library` keyword arguments.
+    """
+    native.cc_library(
+        name = name,
+        copts = tf_copts(),
+        tags = tags,
+        **kwargs
+    )
+
+def tff_cc_binary(name, tags = [], **kwargs):
+    """A version of `cc_binary` that adds default copts.
+
+    It accepts all `cc_binary` arguments.
+
+    Args:
+      name: A unique name for this target.
+      tags: List of arbitrary text tags.
+      **kwargs: `cc_binary` keyword arguments.
+    """
+    native.cc_binary(
+        name = name,
+        copts = tf_copts(),
+        tags = tags,
+        **kwargs
+    )
+
+def tff_cc_test(name, tags = [], **kwargs):
+    """A version of `cc_test` that adds default copts.
+
+    It accepts all `cc_test` arguments.
+
+    Args:
+      name: A unique name for this target.
+      tags: List of arbitrary text tags.
+      **kwargs: `cc_test` keyword arguments.
+    """
+    native.cc_test(
+        name = name,
+        copts = tf_copts(),
+        tags = tags,
+        **kwargs
+    )