chore: import upstream snapshot with attribution

third_party/py/BUILD.bazel

@@ -0,0 +1,74 @@
+load("@bazel_skylib//rules:common_settings.bzl", "bool_flag")
+load("@python_version_repo//:py_version.bzl", "REQUIREMENTS")
+load("@rules_python//python:defs.bzl", "py_binary", "py_library")
+load("@rules_python//python:pip.bzl", "compile_pip_requirements")
+
+compile_pip_requirements(
+    name = "requirements",
+    extra_args = [
+        "--allow-unsafe",
+        "--build-isolation",
+    ],
+    generate_hashes = True,
+    requirements_in = "requirements.in",
+    requirements_txt = REQUIREMENTS,
+)
+
+compile_pip_requirements(
+    name = "requirements_nightly",
+    data = ["test-requirements.txt"],
+    extra_args = [
+        "--allow-unsafe",
+        "--build-isolation",
+        "--extra-index-url=https://pypi.anaconda.org/scientific-python-nightly-wheels/simple",
+        "--pre",
+        "--upgrade",
+    ],
+    generate_hashes = False,
+    requirements_in = "requirements.in",
+    requirements_txt = REQUIREMENTS,
+)
+
+compile_pip_requirements(
+    name = "requirements_dev",
+    extra_args = [
+        "--allow-unsafe",
+        "--build-isolation",
+        "--upgrade",
+    ],
+    generate_hashes = False,
+    requirements_in = "requirements.in",
+    requirements_txt = REQUIREMENTS,
+)
+
+# Flag indicating if the target requires pre-built wheel.
+bool_flag(
+    name = "wheel_dependency",
+    build_setting_default = False,
+)
+
+config_setting(
+    name = "enable_wheel_dependency",
+    flag_values = {
+        ":wheel_dependency": "True",
+    },
+)
+
+filegroup(
+    name = "manylinux_compliance_test",
+    srcs = ["manylinux_compliance_test.py"],
+    visibility = ["//visibility:public"],
+)
+
+py_binary(
+    name = "unpack_wheel_and_unzip_archive_files",
+    srcs = ["unpack_wheel_and_unzip_archive_files.py"],
+    main = "unpack_wheel_and_unzip_archive_files.py",
+    visibility = ["//visibility:public"],
+)
+
+py_library(
+    name = "setup_py_nvidia_dependencies_util",
+    srcs = ["setup_py_nvidia_dependencies_util.py"],
+    visibility = ["//visibility:public"],
+)

third_party/py/BUILD.tpl

@@ -0,0 +1,45 @@
+licenses(["restricted"])
+
+package(default_visibility = ["//visibility:public"])
+
+# Point both runtimes to the same python binary to ensure we always
+# use the python binary specified by ./configure.py script.
+load("@rules_python//python:py_runtime_pair.bzl", "py_runtime_pair")
+
+py_runtime(
+    name = "py3_runtime",
+    interpreter = "%{PYTHON_INTERPRETER}",
+    python_version = "PY3",
+)
+
+py_runtime_pair(
+    name = "py_runtime_pair",
+    py3_runtime = ":py3_runtime",
+)
+
+toolchain(
+    name = "py_toolchain",
+    toolchain = ":py_runtime_pair",
+    toolchain_type = "@bazel_tools//tools/python:toolchain_type",
+    target_compatible_with = [%{PLATFORM_CONSTRAINT}],
+    exec_compatible_with = [%{PLATFORM_CONSTRAINT}],
+)
+
+alias(name = "python_headers",
+      actual = "@rules_python//python/cc:current_py_cc_headers")
+
+# This alias is exists for the use of targets in the @llvm-project dependency,
+# which expect a python_headers target called @python_runtime//:headers. We use
+# a repo_mapping to alias python_runtime to this package, and an alias to create
+# the correct target.
+alias(
+    name = "headers",
+    actual = ":python_headers",
+)
+
+
+config_setting(
+    name = "windows",
+    values = {"cpu": "x64_windows"},
+    visibility = ["//visibility:public"],
+)

third_party/py/absl_py/BUILD.bazel

@@ -0,0 +1 @@
+# copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])

third_party/py/absl_py/workspace.bzl

@@ -0,0 +1,11 @@
+"""Loads the abseil-py library, used by TF."""
+
+load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
+
+def repo():
+    tf_http_archive(
+        name = "absl_py",
+        sha256 = "8a3d0830e4eb4f66c4fa907c06edf6ce1c719ced811a12e26d9d3162f8471758",
+        strip_prefix = "abseil-py-2.1.0",
+        urls = tf_mirror_urls("https://github.com/abseil/abseil-py/archive/refs/tags/v2.1.0.tar.gz"),
+    )

third_party/py/ml_dtypes/ml_dtypes.BUILD

@@ -0,0 +1,60 @@
+""" Main ml_dtypes library. """
+
+load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+
+package(
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],
+)
+
+cc_library(
+    name = "float8",
+    hdrs = ["include/float8.h"],
+    deps = ["@eigen_archive//:eigen3"],
+)
+
+cc_library(
+    name = "intn",
+    hdrs = ["include/intn.h"],
+)
+
+cc_library(
+    name = "mxfloat",
+    hdrs = ["include/mxfloat.h"],
+    deps = [
+        ":float8",
+        "@eigen_archive//:eigen3",
+    ],
+)
+
+pybind_extension(
+    name = "_ml_dtypes_ext",
+    srcs = [
+        "_src/common.h",
+        "_src/custom_float.h",
+        "_src/dtypes.cc",
+        "_src/intn_numpy.h",
+        "_src/numpy.cc",
+        "_src/numpy.h",
+        "_src/ufuncs.h",
+    ],
+    visibility = [":__subpackages__"],
+    deps = [
+        ":float8",
+        ":intn",
+        ":mxfloat",
+        "@eigen_archive//:eigen3",
+        "@xla//third_party/py/numpy:headers",
+    ],
+)
+
+py_library(
+    name = "ml_dtypes",
+    srcs = [
+        "__init__.py",
+        "_finfo.py",
+        "_iinfo.py",
+    ],
+    data = [":_ml_dtypes_ext"],
+    imports = ["."],  # Import relative to _this_ directory, not the root.
+)

third_party/py/ml_dtypes/ml_dtypes_py.BUILD

@@ -0,0 +1,8 @@
+""" Root ml_dtypes_py package. """
+
+package(
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],
+)
+
+exports_files(["LICENSE"])

third_party/py/ml_dtypes/workspace.bzl

@@ -0,0 +1,21 @@
+"""Provides the repo macro to import ml_dtypes.
+
+ml_dtypes provides machine-learning-specific data-types like bfloat16,
+float8 varieties, and int4.
+"""
+
+load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
+
+def repo():
+    ML_DTYPES_COMMIT = "00d98cd92ade342fef589c0470379abb27baebe9"
+    ML_DTYPES_SHA256 = "f6e5880666661351e6cd084ac4178ddc4dabcde7e9a73722981c0d1500cf5937"
+    tf_http_archive(
+        name = "ml_dtypes_py",
+        build_file = "//third_party/py/ml_dtypes:ml_dtypes_py.BUILD",
+        link_files = {
+            "//third_party/py/ml_dtypes:ml_dtypes.BUILD": "ml_dtypes/BUILD.bazel",
+        },
+        sha256 = ML_DTYPES_SHA256,
+        strip_prefix = "ml_dtypes-{commit}".format(commit = ML_DTYPES_COMMIT),
+        urls = tf_mirror_urls("https://github.com/jax-ml/ml_dtypes/archive/{commit}/ml_dtypes-{commit}.tar.gz".format(commit = ML_DTYPES_COMMIT)),
+    )

third_party/py/non_hermetic/ml_dtypes/LICENSE

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

third_party/py/non_hermetic/ml_dtypes/ml_dtypes.BUILD

@@ -0,0 +1,64 @@
+load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+
+package(
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],
+)
+
+exports_files(["LICENSE"])
+
+cc_library(
+    name = "float8",
+    hdrs = ["include/float8.h"],
+    include_prefix = "ml_dtypes",
+    # Internal headers are all relative to . but other packages
+    # include these headers with the prefix.
+    includes = [
+        ".",
+        "ml_dtypes",
+    ],
+    deps = ["@eigen_archive//:eigen3"],
+)
+
+cc_library(
+    name = "intn",
+    hdrs = ["include/intn.h"],
+    include_prefix = "ml_dtypes",
+    # Internal headers are all relative to . but other packages
+    # include these headers with the  prefix.
+    includes = [
+        ".",
+        "ml_dtypes",
+    ],
+)
+
+pybind_extension(
+    name = "_ml_dtypes_ext",
+    srcs = [
+        "_src/common.h",
+        "_src/custom_float.h",
+        "_src/dtypes.cc",
+        "_src/int4_numpy.h",
+        "_src/numpy.cc",
+        "_src/numpy.h",
+        "_src/ufuncs.h",
+    ],
+    includes = ["ml_dtypes"],
+    visibility = [":__subpackages__"],
+    deps = [
+        ":float8",
+        ":intn",
+        "@eigen_archive//:eigen3",
+        "@xla//third_party/py/numpy:headers",
+    ],
+)
+
+py_library(
+    name = "ml_dtypes",
+    srcs = [
+        "__init__.py",
+        "_finfo.py",
+        "_iinfo.py",
+    ],
+    deps = [":_ml_dtypes_ext"],
+)

third_party/py/non_hermetic/ml_dtypes/ml_dtypes.tests.BUILD

@@ -0,0 +1,71 @@
+package(
+    default_visibility = ["//visibility:public"],
+)
+
+py_library(
+    name = "testing_base",
+    deps = [
+        "//:ml_dtypes",
+        "@absl_py//absl/testing:absltest",
+        "@absl_py//absl/testing:parameterized",
+        "@xla//third_party/py/numpy",
+    ],
+)
+
+py_test(
+    name = "custom_float_test",
+    srcs = ["custom_float_test.py"],
+    main = "custom_float_test.py",
+    deps = [":testing_base"],
+)
+
+py_test(
+    name = "int4_test",
+    srcs = ["int4_test.py"],
+    main = "int4_test.py",
+    deps = [":testing_base"],
+)
+
+py_test(
+    name = "iinfo_test",
+    srcs = ["iinfo_test.py"],
+    main = "iinfo_test.py",
+    deps = [":testing_base"],
+)
+
+py_test(
+    name = "finfo_test",
+    srcs = ["finfo_test.py"],
+    main = "finfo_test.py",
+    deps = [":testing_base"],
+)
+
+py_test(
+    name = "metadata_test",
+    srcs = ["metadata_test.py"],
+    main = "metadata_test.py",
+    deps = [":testing_base"],
+)
+
+cc_test(
+    name = "float8_test",
+    srcs = ["float8_test.cc"],
+    linkstatic = 1,
+    deps = [
+        "//:float8",
+        "@com_google_absl//absl/strings",
+        "@com_google_googletest//:gtest_main",
+        "@eigen_archive//:eigen3",
+    ],
+)
+
+cc_test(
+    name = "intn_test_cc",
+    srcs = ["intn_test.cc"],
+    linkstatic = 1,
+    deps = [
+        "//:intn",
+        "@com_google_googletest//:gtest_main",
+        "@eigen_archive//:eigen3",
+    ],
+)

third_party/py/non_hermetic/ml_dtypes/workspace.bzl

@@ -0,0 +1,22 @@
+"""Provides the repo macro to import ml_dtypes.
+
+ml_dtypes provides machine-learning-specific data-types like bfloat16,
+float8 varieties, and int4.
+"""
+
+load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
+
+def repo():
+    ML_DTYPES_COMMIT = "24084d9ed2c3d45bf83b7a9bff833aa185bf9172"
+    ML_DTYPES_SHA256 = "c916a3e6b3d9bdcb476f506fdbbecb6d5e9f21f82f221dfcb42b320b4e85e55a"
+    tf_http_archive(
+        name = "ml_dtypes",
+        build_file = "//third_party/py/ml_dtypes:ml_dtypes.BUILD",
+        link_files = {
+            "//third_party/py/ml_dtypes:ml_dtypes.tests.BUILD": "tests/BUILD.bazel",
+            "//third_party/py/ml_dtypes:LICENSE": "LICENSE",
+        },
+        sha256 = ML_DTYPES_SHA256,
+        strip_prefix = "ml_dtypes-{commit}/ml_dtypes".format(commit = ML_DTYPES_COMMIT),
+        urls = tf_mirror_urls("https://github.com/jax-ml/ml_dtypes/archive/{commit}/ml_dtypes-{commit}.tar.gz".format(commit = ML_DTYPES_COMMIT)),
+    )

third_party/py/non_hermetic/numpy/BUILD.bazel

@@ -0,0 +1,21 @@
+licenses(["restricted"])
+
+package(default_visibility = ["//visibility:public"])
+
+py_library(
+    name = "numpy",
+    srcs = ["tf_numpy_dummy.py"],
+    srcs_version = "PY3",
+)
+
+alias(
+    name = "headers",
+    actual = "@local_config_python//:numpy_headers",
+)
+
+genrule(
+    name = "dummy",
+    outs = ["tf_numpy_dummy.py"],
+    cmd = "touch $@",
+    visibility = ["//visibility:private"],
+)

third_party/py/non_hermetic/numpy/README.md

@@ -0,0 +1,4 @@
+# numpy_ops
+
+The folder tf_numpy_api/ contains lists of NumPy API symbols that the
+`numpy_ops` internal module in TensorFlow implements.

third_party/py/non_hermetic/numpy/tf_numpy_api/BUILD

@@ -0,0 +1,12 @@
+# TensorFlow API backwards compatibility test goldens for tf.experimental.numpy.
+
+package(
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],
+)
+
+filegroup(
+    name = "api_golden",
+    srcs = glob(["*.pbtxt"]),
+)

third_party/py/non_hermetic/numpy/tf_numpy_api/tensorflow.experimental.numpy.ndarray.pbtxt

@@ -0,0 +1,51 @@
+path: "tensorflow.experimental.numpy.ndarray"
+tf_class {
+  is_instance: "<class \'tensorflow.python.framework.tensor.Tensor\'>"
+  is_instance: "<class \'tensorflow.python.types.internal.NativeObject\'>"
+  is_instance: "<class \'tensorflow.python.types.core.Symbol\'>"
+  is_instance: "<class \'tensorflow.python.types.core.Tensor\'>"
+  is_instance: "<type \'object\'>"
+  member {
+    name: "OVERLOADABLE_OPERATORS"
+    mtype: "<type \'set\'>"
+  }
+  member {
+    name: "dtype"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "ndim"
+    mtype: "<type \'property\'>"
+  }
+  member {
+    name: "shape"
+    mtype: "<type \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+  }
+  member_method {
+    name: "eval"
+    argspec: "args=[\'self\', \'feed_dict\', \'session\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "experimental_ref"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "get_shape"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ref"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "set_shape"
+    argspec: "args=[\'self\', \'shape\'], varargs=None, keywords=None, defaults=None"
+  }
+}

third_party/py/non_hermetic/numpy/tf_numpy_api/tensorflow.experimental.numpy.pbtxt

@@ -0,0 +1,919 @@
+path: "tensorflow.experimental.numpy"
+tf_module {
+  member {
+    name: "bool_"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "complex128"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "complex64"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "complex_"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "e"
+    mtype: "<class \'float\'>"
+  }
+  member {
+    name: "float16"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "float32"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "float64"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "float_"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "iinfo"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "inexact"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "inf"
+    mtype: "<class \'float\'>"
+  }
+  member {
+    name: "int16"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "int32"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "int64"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "int8"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "int_"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "ndarray"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "newaxis"
+    mtype: "<type \'NoneType\'>"
+  }
+  member {
+    name: "object_"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "pi"
+    mtype: "<class \'float\'>"
+  }
+  member {
+    name: "random"
+    mtype: "<type \'module\'>"
+  }
+  member {
+    name: "string_"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "uint16"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "uint32"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "uint64"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "uint8"
+    mtype: "<type \'type\'>"
+  }
+  member {
+    name: "unicode_"
+    mtype: "<type \'type\'>"
+  }
+  member_method {
+    name: "abs"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "absolute"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "add"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "all"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "allclose"
+    argspec: "args=[\'a\', \'b\', \'rtol\', \'atol\', \'equal_nan\'], varargs=None, keywords=None, defaults=[\'1e-05\', \'1e-08\', \'False\'], "
+  }
+  member_method {
+    name: "amax"
+    argspec: "args=[\'a\', \'axis\', \'out\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "amin"
+    argspec: "args=[\'a\', \'axis\', \'out\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "angle"
+    argspec: "args=[\'z\', \'deg\'], varargs=None, keywords=None, defaults=[\'False\'], "
+  }
+  member_method {
+    name: "any"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "append"
+    argspec: "args=[\'arr\', \'values\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "arange"
+    argspec: "args=[\'start\', \'stop\', \'step\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'1\', \'None\'], "
+  }
+  member_method {
+    name: "arccos"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arccosh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arcsin"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arcsinh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arctan"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arctan2"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arctanh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "argmax"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "argmin"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "argsort"
+    argspec: "args=[\'a\', \'axis\', \'kind\', \'order\'], varargs=None, keywords=None, defaults=[\'-1\', \'quicksort\', \'None\'], "
+  }
+  member_method {
+    name: "around"
+    argspec: "args=[\'a\', \'decimals\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "array"
+    argspec: "args=[\'val\', \'dtype\', \'copy\', \'ndmin\'], varargs=None, keywords=None, defaults=[\'None\', \'True\', \'0\'], "
+  }
+  member_method {
+    name: "array_equal"
+    argspec: "args=[\'a1\', \'a2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "asanyarray"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "asarray"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "ascontiguousarray"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "atleast_1d"
+    argspec: "args=[], varargs=arys, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "atleast_2d"
+    argspec: "args=[], varargs=arys, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "atleast_3d"
+    argspec: "args=[], varargs=arys, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "average"
+    argspec: "args=[\'a\', \'axis\', \'weights\', \'returned\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\'], "
+  }
+  member_method {
+    name: "bitwise_and"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "bitwise_not"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "bitwise_or"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "bitwise_xor"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "broadcast_arrays"
+    argspec: "args=[], varargs=args, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "broadcast_to"
+    argspec: "args=[\'array\', \'shape\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "cbrt"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ceil"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "clip"
+    argspec: "args=[\'a\', \'a_min\', \'a_max\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "compress"
+    argspec: "args=[\'condition\', \'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "concatenate"
+    argspec: "args=[\'arys\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "conj"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "conjugate"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "copy"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "cos"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "cosh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "count_nonzero"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "cross"
+    argspec: "args=[\'a\', \'b\', \'axisa\', \'axisb\', \'axisc\', \'axis\'], varargs=None, keywords=None, defaults=[\'-1\', \'-1\', \'-1\', \'None\'], "
+  }
+  member_method {
+    name: "cumprod"
+    argspec: "args=[\'a\', \'axis\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "cumsum"
+    argspec: "args=[\'a\', \'axis\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "deg2rad"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "diag"
+    argspec: "args=[\'v\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "diag_indices"
+    argspec: "args=[\'n\', \'ndim\'], varargs=None, keywords=None, defaults=[\'2\'], "
+  }
+  member_method {
+    name: "diagflat"
+    argspec: "args=[\'v\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "diagonal"
+    argspec: "args=[\'a\', \'offset\', \'axis1\', \'axis2\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'1\'], "
+  }
+  member_method {
+    name: "diff"
+    argspec: "args=[\'a\', \'n\', \'axis\'], varargs=None, keywords=None, defaults=[\'1\', \'-1\'], "
+  }
+  member_method {
+    name: "divide"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "divmod"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "dot"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "dsplit"
+    argspec: "args=[\'ary\', \'indices_or_sections\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "dstack"
+    argspec: "args=[\'tup\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "einsum"
+    argspec: "args=[\'subscripts\'], varargs=operands, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "empty"
+    argspec: "args=[\'shape\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "empty_like"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "exp"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "exp2"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "expand_dims"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "experimental_enable_numpy_behavior"
+    argspec: "args=[\'prefer_float32\'], varargs=None, keywords=None, defaults=[\'False\'], "
+  }
+  member_method {
+    name: "expm1"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "eye"
+    argspec: "args=[\'N\', \'M\', \'k\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'0\', \"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "fabs"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "finfo"
+    argspec: "args=[\'dtype\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "fix"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "flatten"
+    argspec: "args=[\'a\', \'order\'], varargs=None, keywords=None, defaults=[\'C\'], "
+  }
+  member_method {
+    name: "flip"
+    argspec: "args=[\'m\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "fliplr"
+    argspec: "args=[\'m\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "flipud"
+    argspec: "args=[\'m\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "float_power"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "floor"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "floor_divide"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "full"
+    argspec: "args=[\'shape\', \'fill_value\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "full_like"
+    argspec: "args=[\'a\', \'fill_value\', \'dtype\', \'order\', \'subok\', \'shape\'], varargs=None, keywords=None, defaults=[\'None\', \'K\', \'True\', \'None\'], "
+  }
+  member_method {
+    name: "gcd"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "geomspace"
+    argspec: "args=[\'start\', \'stop\', \'num\', \'endpoint\', \'dtype\', \'axis\'], varargs=None, keywords=None, defaults=[\'50\', \'True\', \'None\', \'0\'], "
+  }
+  member_method {
+    name: "greater"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "greater_equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "heaviside"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "hsplit"
+    argspec: "args=[\'ary\', \'indices_or_sections\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "hstack"
+    argspec: "args=[\'tup\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "hypot"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "identity"
+    argspec: "args=[\'n\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "imag"
+    argspec: "args=[\'val\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "inner"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isclose"
+    argspec: "args=[\'a\', \'b\', \'rtol\', \'atol\', \'equal_nan\'], varargs=None, keywords=None, defaults=[\'1e-05\', \'1e-08\', \'False\'], "
+  }
+  member_method {
+    name: "iscomplex"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "iscomplexobj"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isfinite"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isinf"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isnan"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isneginf"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isposinf"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isreal"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isrealobj"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isscalar"
+    argspec: "args=[\'num\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "issubdtype"
+    argspec: "args=[\'arg1\', \'arg2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ix_"
+    argspec: "args=[], varargs=args, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "kron"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "lcm"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "less"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "less_equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "linspace"
+    argspec: "args=[\'start\', \'stop\', \'num\', \'endpoint\', \'retstep\', \'dtype\', \'axis\'], varargs=None, keywords=None, defaults=[\'50\', \'True\', \'False\', \"<class \'float\'>\", \'0\'], "
+  }
+  member_method {
+    name: "log"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "log10"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "log1p"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "log2"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logaddexp"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logaddexp2"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_and"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_not"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_or"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_xor"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logspace"
+    argspec: "args=[\'start\', \'stop\', \'num\', \'endpoint\', \'base\', \'dtype\', \'axis\'], varargs=None, keywords=None, defaults=[\'50\', \'True\', \'10.0\', \'None\', \'0\'], "
+  }
+  member_method {
+    name: "matmul"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "max"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "maximum"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "mean"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'out\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "meshgrid"
+    argspec: "args=[], varargs=xi, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "min"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "minimum"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "mod"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "moveaxis"
+    argspec: "args=[\'a\', \'source\', \'destination\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "multiply"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "nanmean"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "nanprod"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\'], "
+  }
+  member_method {
+    name: "nansum"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\'], "
+  }
+  member_method {
+    name: "ndim"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "negative"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "nextafter"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "nonzero"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "not_equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ones"
+    argspec: "args=[\'shape\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "ones_like"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "outer"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "pad"
+    argspec: "args=[\'array\', \'pad_width\', \'mode\'], varargs=None, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "polyval"
+    argspec: "args=[\'p\', \'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "positive"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "power"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "prod"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "promote_types"
+    argspec: "args=[\'type1\', \'type2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ptp"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "rad2deg"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ravel"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "real"
+    argspec: "args=[\'val\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "reciprocal"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "remainder"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "repeat"
+    argspec: "args=[\'a\', \'repeats\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "reshape"
+    argspec: "args=[\'a\', \'newshape\', \'order\'], varargs=None, keywords=None, defaults=[\'C\'], "
+  }
+  member_method {
+    name: "result_type"
+    argspec: "args=[], varargs=arrays_and_dtypes, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "roll"
+    argspec: "args=[\'a\', \'shift\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "rot90"
+    argspec: "args=[\'m\', \'k\', \'axes\'], varargs=None, keywords=None, defaults=[\'1\', \'(0, 1)\'], "
+  }
+  member_method {
+    name: "round"
+    argspec: "args=[\'a\', \'decimals\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "select"
+    argspec: "args=[\'condlist\', \'choicelist\', \'default\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "shape"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sign"
+    argspec: "args=[\'x\', \'out\', \'where\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "signbit"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sin"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sinc"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sinh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "size"
+    argspec: "args=[\'x\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "sort"
+    argspec: "args=[\'a\', \'axis\', \'kind\', \'order\'], varargs=None, keywords=None, defaults=[\'-1\', \'quicksort\', \'None\'], "
+  }
+  member_method {
+    name: "split"
+    argspec: "args=[\'ary\', \'indices_or_sections\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "sqrt"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "square"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "squeeze"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "stack"
+    argspec: "args=[\'arrays\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "std"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "subtract"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sum"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "swapaxes"
+    argspec: "args=[\'a\', \'axis1\', \'axis2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "take"
+    argspec: "args=[\'a\', \'indices\', \'axis\', \'out\', \'mode\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'clip\'], "
+  }
+  member_method {
+    name: "take_along_axis"
+    argspec: "args=[\'arr\', \'indices\', \'axis\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "tan"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "tanh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "tensordot"
+    argspec: "args=[\'a\', \'b\', \'axes\'], varargs=None, keywords=None, defaults=[\'2\'], "
+  }
+  member_method {
+    name: "tile"
+    argspec: "args=[\'a\', \'reps\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "trace"
+    argspec: "args=[\'a\', \'offset\', \'axis1\', \'axis2\', \'dtype\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'1\', \'None\'], "
+  }
+  member_method {
+    name: "transpose"
+    argspec: "args=[\'a\', \'axes\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "tri"
+    argspec: "args=[\'N\', \'M\', \'k\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'0\', \'None\'], "
+  }
+  member_method {
+    name: "tril"
+    argspec: "args=[\'m\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "triu"
+    argspec: "args=[\'m\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "true_divide"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "vander"
+    argspec: "args=[\'x\', \'N\', \'increasing\'], varargs=None, keywords=None, defaults=[\'None\', \'False\'], "
+  }
+  member_method {
+    name: "var"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'out\', \'ddof\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'0\', \'None\'], "
+  }
+  member_method {
+    name: "vdot"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "vsplit"
+    argspec: "args=[\'ary\', \'indices_or_sections\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "vstack"
+    argspec: "args=[\'tup\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "where"
+    argspec: "args=[\'condition\', \'x\', \'y\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "zeros"
+    argspec: "args=[\'shape\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "zeros_like"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

third_party/py/non_hermetic/numpy/tf_numpy_api/tensorflow.experimental.numpy.random.pbtxt

@@ -0,0 +1,35 @@
+path: "tensorflow.experimental.numpy.random"
+tf_module {
+  member_method {
+    name: "poisson"
+    argspec: "args=[\'lam\', \'size\'], varargs=None, keywords=None, defaults=[\'1.0\', \'None\'], "
+  }
+  member_method {
+    name: "rand"
+    argspec: "args=[], varargs=size, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "randint"
+    argspec: "args=[\'low\', \'high\', \'size\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \"<class \'numpy.int64\'>\"], "
+  }
+  member_method {
+    name: "randn"
+    argspec: "args=[], varargs=args, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "random"
+    argspec: "args=[\'size\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "seed"
+    argspec: "args=[\'s\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "standard_normal"
+    argspec: "args=[\'size\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "uniform"
+    argspec: "args=[\'low\', \'high\', \'size\'], varargs=None, keywords=None, defaults=[\'0.0\', \'1.0\', \'None\'], "
+  }
+}

third_party/py/numpy/BUILD.bazel

@@ -0,0 +1,20 @@
+licenses(["restricted"])
+
+package(default_visibility = ["//visibility:public"])
+
+alias(
+    name = "numpy",
+    actual = "@pypi//numpy",
+)
+
+alias(
+    name = "headers",
+    actual = "@pypi//numpy:numpy_headers",
+)
+
+genrule(
+    name = "dummy",
+    outs = ["tf_numpy_dummy.py"],
+    cmd = "touch $@",
+    visibility = ["//visibility:private"],
+)

third_party/py/numpy/LICENSE

@@ -0,0 +1,60 @@
+Copyright (c) 2005-2019, NumPy Developers.
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are
+met:
+
+    * Redistributions of source code must retain the above copyright
+       notice, this list of conditions and the following disclaimer.
+
+    * Redistributions in binary form must reproduce the above
+       copyright notice, this list of conditions and the following
+       disclaimer in the documentation and/or other materials provided
+       with the distribution.
+
+    * Neither the name of the NumPy Developers nor the names of any
+       contributors may be used to endorse or promote products derived
+       from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+
+
+The NumPy repository and source distributions bundle several libraries that are
+compatibly licensed.  We list these here.
+
+Name: Numpydoc
+Files: doc/sphinxext/numpydoc/*
+License: 2-clause BSD
+  For details, see doc/sphinxext/LICENSE.txt
+
+Name: scipy-sphinx-theme
+Files: doc/scipy-sphinx-theme/*
+License: 3-clause BSD, PSF and Apache 2.0
+  For details, see doc/scipy-sphinx-theme/LICENSE.txt
+
+Name: lapack-lite
+Files: numpy/linalg/lapack_lite/*
+License: 3-clause BSD
+  For details, see numpy/linalg/lapack_lite/LICENSE.txt
+
+Name: tempita
+Files: tools/npy_tempita/*
+License: BSD derived
+  For details, see tools/npy_tempita/license.txt
+
+Name: dragon4
+Files: numpy/core/src/multiarray/dragon4.c
+License: One of a kind
+  For license text, see numpy/core/src/multiarray/dragon4.c

third_party/py/numpy/README.md

@@ -0,0 +1,4 @@
+# numpy_ops
+
+The folder tf_numpy_api/ contains lists of NumPy API symbols that the
+`numpy_ops` internal module in TensorFlow implements.

third_party/py/numpy/tf_numpy_api/BUILD

@@ -0,0 +1,12 @@
+# TensorFlow API backwards compatibility test goldens for tf.experimental.numpy.
+
+package(
+    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+    default_visibility = ["//visibility:public"],
+    licenses = ["notice"],
+)
+
+filegroup(
+    name = "api_golden",
+    srcs = glob(["*.pbtxt"]),
+)

third_party/py/numpy/tf_numpy_api/tensorflow.experimental.numpy.ndarray.pbtxt

@@ -0,0 +1,51 @@
+path: "tensorflow.experimental.numpy.ndarray"
+tf_class {
+  is_instance: "<class \'tensorflow.python.framework.tensor.Tensor\'>"
+  is_instance: "<class \'tensorflow.python.types.internal.NativeObject\'>"
+  is_instance: "<class \'tensorflow.python.types.core.Symbol\'>"
+  is_instance: "<class \'tensorflow.python.types.core.Tensor\'>"
+  is_instance: "<class \'object\'>"
+  member {
+    name: "OVERLOADABLE_OPERATORS"
+    mtype: "<class \'set\'>"
+  }
+  member {
+    name: "dtype"
+    mtype: "<class \'property\'>"
+  }
+  member {
+    name: "name"
+    mtype: "<class \'property\'>"
+  }
+  member {
+    name: "ndim"
+    mtype: "<class \'property\'>"
+  }
+  member {
+    name: "shape"
+    mtype: "<class \'property\'>"
+  }
+  member_method {
+    name: "__init__"
+  }
+  member_method {
+    name: "eval"
+    argspec: "args=[\'self\', \'feed_dict\', \'session\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "experimental_ref"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "get_shape"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ref"
+    argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "set_shape"
+    argspec: "args=[\'self\', \'shape\'], varargs=None, keywords=None, defaults=None"
+  }
+}

third_party/py/numpy/tf_numpy_api/tensorflow.experimental.numpy.pbtxt

@@ -0,0 +1,919 @@
+path: "tensorflow.experimental.numpy"
+tf_module {
+  member {
+    name: "bool_"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "complex128"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "complex64"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "complex_"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "e"
+    mtype: "<class \'float\'>"
+  }
+  member {
+    name: "float16"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "float32"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "float64"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "float_"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "iinfo"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "inexact"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "inf"
+    mtype: "<class \'float\'>"
+  }
+  member {
+    name: "int16"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "int32"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "int64"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "int8"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "int_"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "ndarray"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "newaxis"
+    mtype: "<class \'NoneType\'>"
+  }
+  member {
+    name: "object_"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "pi"
+    mtype: "<class \'float\'>"
+  }
+  member {
+    name: "random"
+    mtype: "<class \'module\'>"
+  }
+  member {
+    name: "string_"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "uint16"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "uint32"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "uint64"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "uint8"
+    mtype: "<class \'type\'>"
+  }
+  member {
+    name: "unicode_"
+    mtype: "<class \'type\'>"
+  }
+  member_method {
+    name: "abs"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "absolute"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "add"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "all"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "allclose"
+    argspec: "args=[\'a\', \'b\', \'rtol\', \'atol\', \'equal_nan\'], varargs=None, keywords=None, defaults=[\'1e-05\', \'1e-08\', \'False\'], "
+  }
+  member_method {
+    name: "amax"
+    argspec: "args=[\'a\', \'axis\', \'out\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "amin"
+    argspec: "args=[\'a\', \'axis\', \'out\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "angle"
+    argspec: "args=[\'z\', \'deg\'], varargs=None, keywords=None, defaults=[\'False\'], "
+  }
+  member_method {
+    name: "any"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "append"
+    argspec: "args=[\'arr\', \'values\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "arange"
+    argspec: "args=[\'start\', \'stop\', \'step\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'1\', \'None\'], "
+  }
+  member_method {
+    name: "arccos"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arccosh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arcsin"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arcsinh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arctan"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arctan2"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "arctanh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "argmax"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "argmin"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "argsort"
+    argspec: "args=[\'a\', \'axis\', \'kind\', \'order\'], varargs=None, keywords=None, defaults=[\'-1\', \'quicksort\', \'None\'], "
+  }
+  member_method {
+    name: "around"
+    argspec: "args=[\'a\', \'decimals\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "array"
+    argspec: "args=[\'val\', \'dtype\', \'copy\', \'ndmin\'], varargs=None, keywords=None, defaults=[\'None\', \'True\', \'0\'], "
+  }
+  member_method {
+    name: "array_equal"
+    argspec: "args=[\'a1\', \'a2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "asanyarray"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "asarray"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "ascontiguousarray"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "atleast_1d"
+    argspec: "args=[], varargs=arys, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "atleast_2d"
+    argspec: "args=[], varargs=arys, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "atleast_3d"
+    argspec: "args=[], varargs=arys, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "average"
+    argspec: "args=[\'a\', \'axis\', \'weights\', \'returned\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\'], "
+  }
+  member_method {
+    name: "bitwise_and"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "bitwise_not"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "bitwise_or"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "bitwise_xor"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "broadcast_arrays"
+    argspec: "args=[], varargs=args, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "broadcast_to"
+    argspec: "args=[\'array\', \'shape\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "cbrt"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ceil"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "clip"
+    argspec: "args=[\'a\', \'a_min\', \'a_max\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "compress"
+    argspec: "args=[\'condition\', \'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "concatenate"
+    argspec: "args=[\'arys\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "conj"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "conjugate"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "copy"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "cos"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "cosh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "count_nonzero"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "cross"
+    argspec: "args=[\'a\', \'b\', \'axisa\', \'axisb\', \'axisc\', \'axis\'], varargs=None, keywords=None, defaults=[\'-1\', \'-1\', \'-1\', \'None\'], "
+  }
+  member_method {
+    name: "cumprod"
+    argspec: "args=[\'a\', \'axis\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "cumsum"
+    argspec: "args=[\'a\', \'axis\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "deg2rad"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "diag"
+    argspec: "args=[\'v\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "diag_indices"
+    argspec: "args=[\'n\', \'ndim\'], varargs=None, keywords=None, defaults=[\'2\'], "
+  }
+  member_method {
+    name: "diagflat"
+    argspec: "args=[\'v\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "diagonal"
+    argspec: "args=[\'a\', \'offset\', \'axis1\', \'axis2\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'1\'], "
+  }
+  member_method {
+    name: "diff"
+    argspec: "args=[\'a\', \'n\', \'axis\'], varargs=None, keywords=None, defaults=[\'1\', \'-1\'], "
+  }
+  member_method {
+    name: "divide"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "divmod"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "dot"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "dsplit"
+    argspec: "args=[\'ary\', \'indices_or_sections\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "dstack"
+    argspec: "args=[\'tup\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "einsum"
+    argspec: "args=[\'subscripts\'], varargs=operands, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "empty"
+    argspec: "args=[\'shape\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "empty_like"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "exp"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "exp2"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "expand_dims"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "experimental_enable_numpy_behavior"
+    argspec: "args=[\'prefer_float32\', \'dtype_conversion_mode\'], varargs=None, keywords=None, defaults=[\'False\', \'legacy\'], "
+  }
+  member_method {
+    name: "expm1"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "eye"
+    argspec: "args=[\'N\', \'M\', \'k\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'0\', \"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "fabs"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "finfo"
+    argspec: "args=[\'dtype\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "fix"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "flatten"
+    argspec: "args=[\'a\', \'order\'], varargs=None, keywords=None, defaults=[\'C\'], "
+  }
+  member_method {
+    name: "flip"
+    argspec: "args=[\'m\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "fliplr"
+    argspec: "args=[\'m\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "flipud"
+    argspec: "args=[\'m\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "float_power"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "floor"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "floor_divide"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "full"
+    argspec: "args=[\'shape\', \'fill_value\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "full_like"
+    argspec: "args=[\'a\', \'fill_value\', \'dtype\', \'order\', \'subok\', \'shape\'], varargs=None, keywords=None, defaults=[\'None\', \'K\', \'True\', \'None\'], "
+  }
+  member_method {
+    name: "gcd"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "geomspace"
+    argspec: "args=[\'start\', \'stop\', \'num\', \'endpoint\', \'dtype\', \'axis\'], varargs=None, keywords=None, defaults=[\'50\', \'True\', \'None\', \'0\'], "
+  }
+  member_method {
+    name: "greater"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "greater_equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "heaviside"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "hsplit"
+    argspec: "args=[\'ary\', \'indices_or_sections\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "hstack"
+    argspec: "args=[\'tup\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "hypot"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "identity"
+    argspec: "args=[\'n\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "imag"
+    argspec: "args=[\'val\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "inner"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isclose"
+    argspec: "args=[\'a\', \'b\', \'rtol\', \'atol\', \'equal_nan\'], varargs=None, keywords=None, defaults=[\'1e-05\', \'1e-08\', \'False\'], "
+  }
+  member_method {
+    name: "iscomplex"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "iscomplexobj"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isfinite"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isinf"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isnan"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isneginf"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isposinf"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isreal"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isrealobj"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "isscalar"
+    argspec: "args=[\'num\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "issubdtype"
+    argspec: "args=[\'arg1\', \'arg2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ix_"
+    argspec: "args=[], varargs=args, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "kron"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "lcm"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "less"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "less_equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "linspace"
+    argspec: "args=[\'start\', \'stop\', \'num\', \'endpoint\', \'retstep\', \'dtype\', \'axis\'], varargs=None, keywords=None, defaults=[\'50\', \'True\', \'False\', \"<class \'float\'>\", \'0\'], "
+  }
+  member_method {
+    name: "log"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "log10"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "log1p"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "log2"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logaddexp"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logaddexp2"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_and"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_not"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_or"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logical_xor"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "logspace"
+    argspec: "args=[\'start\', \'stop\', \'num\', \'endpoint\', \'base\', \'dtype\', \'axis\'], varargs=None, keywords=None, defaults=[\'50\', \'True\', \'10.0\', \'None\', \'0\'], "
+  }
+  member_method {
+    name: "matmul"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "max"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "maximum"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "mean"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'out\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "meshgrid"
+    argspec: "args=[], varargs=xi, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "min"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "minimum"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "mod"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "moveaxis"
+    argspec: "args=[\'a\', \'source\', \'destination\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "multiply"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "nanmean"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "nanprod"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\'], "
+  }
+  member_method {
+    name: "nansum"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\'], "
+  }
+  member_method {
+    name: "ndim"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "negative"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "nextafter"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "nonzero"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "not_equal"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ones"
+    argspec: "args=[\'shape\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "ones_like"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "outer"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "pad"
+    argspec: "args=[\'array\', \'pad_width\', \'mode\'], varargs=None, keywords=kwargs, defaults=None"
+  }
+  member_method {
+    name: "polyval"
+    argspec: "args=[\'p\', \'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "positive"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "power"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "prod"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "promote_types"
+    argspec: "args=[\'type1\', \'type2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ptp"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "rad2deg"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "ravel"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "real"
+    argspec: "args=[\'val\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "reciprocal"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "remainder"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "repeat"
+    argspec: "args=[\'a\', \'repeats\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "reshape"
+    argspec: "args=[\'a\', \'newshape\', \'order\'], varargs=None, keywords=None, defaults=[\'C\'], "
+  }
+  member_method {
+    name: "result_type"
+    argspec: "args=[], varargs=arrays_and_dtypes, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "roll"
+    argspec: "args=[\'a\', \'shift\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "rot90"
+    argspec: "args=[\'m\', \'k\', \'axes\'], varargs=None, keywords=None, defaults=[\'1\', \'(0, 1)\'], "
+  }
+  member_method {
+    name: "round"
+    argspec: "args=[\'a\', \'decimals\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "select"
+    argspec: "args=[\'condlist\', \'choicelist\', \'default\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "shape"
+    argspec: "args=[\'a\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sign"
+    argspec: "args=[\'x\', \'out\', \'where\'], varargs=None, keywords=kwargs, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "signbit"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sin"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sinc"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sinh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "size"
+    argspec: "args=[\'x\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "sort"
+    argspec: "args=[\'a\', \'axis\', \'kind\', \'order\'], varargs=None, keywords=None, defaults=[\'-1\', \'quicksort\', \'None\'], "
+  }
+  member_method {
+    name: "split"
+    argspec: "args=[\'ary\', \'indices_or_sections\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "sqrt"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "square"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "squeeze"
+    argspec: "args=[\'a\', \'axis\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "stack"
+    argspec: "args=[\'arrays\', \'axis\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "std"
+    argspec: "args=[\'a\', \'axis\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "subtract"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "sum"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "swapaxes"
+    argspec: "args=[\'a\', \'axis1\', \'axis2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "take"
+    argspec: "args=[\'a\', \'indices\', \'axis\', \'out\', \'mode\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'clip\'], "
+  }
+  member_method {
+    name: "take_along_axis"
+    argspec: "args=[\'arr\', \'indices\', \'axis\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "tan"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "tanh"
+    argspec: "args=[\'x\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "tensordot"
+    argspec: "args=[\'a\', \'b\', \'axes\'], varargs=None, keywords=None, defaults=[\'2\'], "
+  }
+  member_method {
+    name: "tile"
+    argspec: "args=[\'a\', \'reps\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "trace"
+    argspec: "args=[\'a\', \'offset\', \'axis1\', \'axis2\', \'dtype\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'1\', \'None\'], "
+  }
+  member_method {
+    name: "transpose"
+    argspec: "args=[\'a\', \'axes\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "tri"
+    argspec: "args=[\'N\', \'M\', \'k\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'0\', \'None\'], "
+  }
+  member_method {
+    name: "tril"
+    argspec: "args=[\'m\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "triu"
+    argspec: "args=[\'m\', \'k\'], varargs=None, keywords=None, defaults=[\'0\'], "
+  }
+  member_method {
+    name: "true_divide"
+    argspec: "args=[\'x1\', \'x2\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "vander"
+    argspec: "args=[\'x\', \'N\', \'increasing\'], varargs=None, keywords=None, defaults=[\'None\', \'False\'], "
+  }
+  member_method {
+    name: "var"
+    argspec: "args=[\'a\', \'axis\', \'dtype\', \'out\', \'ddof\', \'keepdims\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'0\', \'None\'], "
+  }
+  member_method {
+    name: "vdot"
+    argspec: "args=[\'a\', \'b\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "vsplit"
+    argspec: "args=[\'ary\', \'indices_or_sections\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "vstack"
+    argspec: "args=[\'tup\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "where"
+    argspec: "args=[\'condition\', \'x\', \'y\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
+  }
+  member_method {
+    name: "zeros"
+    argspec: "args=[\'shape\', \'dtype\'], varargs=None, keywords=None, defaults=[\"<class \'float\'>\"], "
+  }
+  member_method {
+    name: "zeros_like"
+    argspec: "args=[\'a\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+}

third_party/py/numpy/tf_numpy_api/tensorflow.experimental.numpy.random.pbtxt

@@ -0,0 +1,35 @@
+path: "tensorflow.experimental.numpy.random"
+tf_module {
+  member_method {
+    name: "poisson"
+    argspec: "args=[\'lam\', \'size\'], varargs=None, keywords=None, defaults=[\'1.0\', \'None\'], "
+  }
+  member_method {
+    name: "rand"
+    argspec: "args=[], varargs=size, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "randint"
+    argspec: "args=[\'low\', \'high\', \'size\', \'dtype\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \"<class \'numpy.int64\'>\"], "
+  }
+  member_method {
+    name: "randn"
+    argspec: "args=[], varargs=args, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "random"
+    argspec: "args=[\'size\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "seed"
+    argspec: "args=[\'s\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "standard_normal"
+    argspec: "args=[\'size\'], varargs=None, keywords=None, defaults=[\'None\'], "
+  }
+  member_method {
+    name: "uniform"
+    argspec: "args=[\'low\', \'high\', \'size\'], varargs=None, keywords=None, defaults=[\'0.0\', \'1.0\', \'None\'], "
+  }
+}

third_party/py/python_configure.bzl

@@ -0,0 +1,83 @@
+"""Repository rule for Python autoconfiguration.
+"""
+
+load("@python_version_repo//:py_version.bzl", "HERMETIC_PYTHON_VERSION")
+load(
+    "@xla//third_party/remote_config:common.bzl",
+    "BAZEL_SH",
+    "PYTHON_BIN_PATH",
+    "PYTHON_LIB_PATH",
+)
+
+def _get_python_interpreter():
+    python_toolchain_name = "python_{version}_host".format(
+        version = HERMETIC_PYTHON_VERSION.replace(".", "_"),
+    )
+    return "@{}//:python".format(python_toolchain_name)
+
+def _create_local_python_repository(repository_ctx):
+    """Creates the repository containing files set up to build with Python."""
+
+    platform_constraint = ""
+    if repository_ctx.attr.platform_constraint:
+        platform_constraint = "\"%s\"" % repository_ctx.attr.platform_constraint
+    repository_ctx.template(
+        "BUILD",
+        repository_ctx.attr.build_tpl,
+        {
+            "%{PLATFORM_CONSTRAINT}": platform_constraint,
+            "%{PYTHON_INTERPRETER}": repository_ctx.attr.python_interpreter,
+        },
+    )
+
+def _python_autoconf_impl(repository_ctx):
+    """Implementation of the python_autoconf repository rule."""
+    _create_local_python_repository(repository_ctx)
+
+_ENVIRONS = [
+    BAZEL_SH,
+    PYTHON_BIN_PATH,
+    PYTHON_LIB_PATH,
+]
+
+local_python_configure = repository_rule(
+    implementation = _create_local_python_repository,
+    attrs = {
+        "environ": attr.string_dict(),
+        "platform_constraint": attr.string(),
+        "build_tpl": attr.label(default = Label("//third_party/py:BUILD.tpl")),
+        "python_interpreter": attr.string(default = _get_python_interpreter()),
+    },
+)
+
+remote_python_configure = repository_rule(
+    implementation = _create_local_python_repository,
+    environ = _ENVIRONS,
+    remotable = True,
+    attrs = {
+        "environ": attr.string_dict(),
+        "platform_constraint": attr.string(),
+        "build_tpl": attr.label(default = Label("//third_party/py:BUILD.tpl")),
+        "python_interpreter": attr.string(default = _get_python_interpreter()),
+    },
+)
+
+python_configure = repository_rule(
+    implementation = _python_autoconf_impl,
+    attrs = {
+        "platform_constraint": attr.string(),
+        "build_tpl": attr.label(default = Label("//third_party/py:BUILD.tpl")),
+        "python_interpreter": attr.string(default = _get_python_interpreter()),
+    },
+)
+"""Detects and configures the local Python.
+
+Add the following to your WORKSPACE FILE:
+
+```python
+python_configure(name = "local_config_python")
+```
+
+Args:
+  name: A unique name for this workspace rule.
+"""

third_party/py/python_init_pip.bzl

@@ -0,0 +1,13 @@
+"""
+Legacy-compatible wrapper for Hermetic Python initialization.
+See the WORKSPACE file for instructions on using the updated
+Hermetic Python initialization.
+"""
+
+load(
+    "@rules_ml_toolchain//py:python_init_pip.bzl",
+    _python_init_pip = "python_init_pip",
+)
+
+def python_init_pip():
+    _python_init_pip()

third_party/py/python_init_repositories.bzl

@@ -0,0 +1,26 @@
+"""
+Legacy-compatible wrapper for Hermetic Python initialization.
+See the WORKSPACE file for instructions on using the updated
+Hermetic Python initialization.
+"""
+
+load(
+    "@rules_ml_toolchain//py:python_init_repositories.bzl",
+    _python_init_repositories = "python_init_repositories",
+)
+
+def python_init_repositories(
+        requirements = {},
+        local_wheel_workspaces = [],
+        local_wheel_dist_folder = None,
+        default_python_version = None,
+        local_wheel_inclusion_list = ["*"],
+        local_wheel_exclusion_list = []):
+    _python_init_repositories(
+        requirements = requirements,
+        local_wheel_workspaces = local_wheel_workspaces,
+        local_wheel_dist_folder = local_wheel_dist_folder,
+        default_python_version = default_python_version,
+        local_wheel_inclusion_list = local_wheel_inclusion_list,
+        local_wheel_exclusion_list = local_wheel_exclusion_list,
+    )

third_party/py/python_init_rules.bzl

@@ -0,0 +1,39 @@
+"""Hermetic Python initialization. Consult the WORKSPACE on how to use it."""
+
+load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
+
+def python_init_rules(extra_patches = []):
+    """Defines (doesn't setup) the rules_python repository.
+
+    Args:
+      extra_patches: list of labels. Additional patches to apply after the default
+        set of patches.
+    """
+
+    tf_http_archive(
+        name = "com_google_protobuf",
+        patch_file = [
+            "@xla//third_party/protobuf:protobuf.patch",
+            "@xla//third_party/protobuf:protobuf_arena.patch",
+        ],
+        sha256 = "6e09bbc950ba60c3a7b30280210cd285af8d7d8ed5e0a6ed101c72aff22e8d88",
+        strip_prefix = "protobuf-6.31.1",
+        urls = tf_mirror_urls("https://github.com/protocolbuffers/protobuf/archive/refs/tags/v6.31.1.zip"),
+        repo_mapping = {
+            "@abseil-cpp": "@com_google_absl",
+            "@protobuf_pip_deps": "@pypi",
+        },
+    )
+
+    tf_http_archive(
+        name = "rules_python",
+        sha256 = "8964aa1e7525fea5244ba737458694a057ada1be96a92998a41caa1983562d00",
+        strip_prefix = "rules_python-1.8.5",
+        urls = tf_mirror_urls("https://github.com/bazelbuild/rules_python/releases/download/1.8.5/rules_python-1.8.5.tar.gz"),
+        patch_file = [
+            "@xla//third_party/py:rules_python_pip_version.patch",
+            "@xla//third_party/py:rules_python_scope.patch",
+            "@xla//third_party/py:rules_python_freethreaded.patch",
+            "@xla//third_party/py:rules_python_versions.patch",
+        ] + extra_patches,
+    )

third_party/py/python_init_toolchains.bzl

@@ -0,0 +1,17 @@
+"""
+Legacy-compatible wrapper for python toolchain registration.
+See the WORKSPACE file for instructions on using the updated
+python toolchain registration.
+"""
+
+load(
+    "@rules_ml_toolchain//py:python_register_toolchain.bzl",
+    "python_register_toolchain",
+)
+
+def python_init_toolchains(name = "python", python_version = None, **kwargs):
+    python_register_toolchain(
+        name = name,
+        python_version = python_version,
+        **kwargs
+    )

third_party/py/python_repo.bzl

@@ -0,0 +1,708 @@
+"""
+Repository rule to manage hermetic Python interpreter under Bazel.
+
+Version can be set via build parameter "--repo_env=HERMETIC_PYTHON_VERSION=3.11"
+
+To set wheel name, add "--repo_env=WHEEL_NAME=tensorflow_cpu"
+"""
+
+DEFAULT_VERSION = "3.11"
+
+def _python_repository_impl(ctx):
+    version, py_kind = _get_python_version(ctx)
+    version_and_kind = "%s-%s" % (version, py_kind) if py_kind else version
+
+    ctx.file("BUILD", "")
+    wheel_name = ctx.os.environ.get("WHEEL_NAME", "tensorflow")
+    wheel_collab = ctx.os.environ.get("WHEEL_COLLAB", False)
+    macos_deployment_target = ctx.os.environ.get("MACOSX_DEPLOYMENT_TARGET", "")
+    hermetic_url = ctx.os.environ.get("HERMETIC_PYTHON_URL", "")
+    hermetic_sha256 = ctx.os.environ.get("HERMETIC_PYTHON_SHA256", "")
+    hermetic_prefix = ctx.os.environ.get("HERMETIC_PYTHON_PREFIX", "python")
+    custom_requirements = ctx.os.environ.get("HERMETIC_REQUIREMENTS_LOCK", None)
+    local_wheel_overrides_label = ""
+
+    if not (hermetic_url + hermetic_sha256) and (hermetic_url or hermetic_sha256):
+        fail("""
+Please either specify both HERMETIC_PYTHON_URL and HERMETIC_PYTHON_SHA256
+to set up a custom python interpreter, or none of them to rely on default ones.
+""")
+    requirements = None
+    if not requirements:
+        for i in range(0, len(ctx.attr.requirements_locks)):
+            if ctx.attr.requirements_versions[i] == version_and_kind:
+                requirements = ctx.attr.requirements_locks[i]
+                break
+
+    if not requirements and not custom_requirements:
+        fail("""
+Could not find requirements_lock.txt file matching specified Python version.
+Specified python version: {version}
+Python versions with available requirement_lock.txt files: {versions}
+Please check python_init_repositories() in your WORKSPACE file.
+""".format(
+            version = version,
+            versions = ", ".join(ctx.attr.requirements_versions),
+        ))
+
+    if custom_requirements:
+        custom_requirements_path = ctx.path(custom_requirements)
+        requirements_with_local_wheels = "@{repo}//:{label}".format(
+            repo = ctx.name,
+            label = custom_requirements_path.basename,
+        )
+        ctx.file(
+            custom_requirements_path.basename,
+            ctx.read(custom_requirements_path),
+        )
+    elif ctx.attr.local_wheel_workspaces:
+        base_requirements = ctx.read(requirements)
+        merged_requirements_blocks, override_report_entries = (
+            _rewrite_requirements_with_local_wheels(
+                ctx,
+                version,
+                ctx.attr.local_wheel_workspaces,
+                base_requirements,
+            )
+        )
+        merged_requirements_content = "\n".join(merged_requirements_blocks)
+
+        requirements_with_local_wheels = "@{repo}//:{label}".format(
+            repo = ctx.name,
+            label = requirements.name,
+        )
+        ctx.file(
+            requirements.name,
+            merged_requirements_content,
+        )
+        local_wheel_overrides_label = _write_local_wheel_overrides_json(
+            ctx,
+            version,
+            py_kind,
+            requirements_with_local_wheels,
+            override_report_entries,
+        )
+        _print_local_wheel_override_summaries(override_report_entries)
+    else:
+        requirements_with_local_wheels = str(requirements)
+
+    use_pywrap_rules = bool(
+        ctx.os.environ.get("USE_PYWRAP_RULES", False),
+    )
+
+    if use_pywrap_rules:
+        print("!!!Using pywrap rules instead of directly creating .so objects!!!")  # buildifier: disable=print
+
+    interpreter_type = "\"default\" (provided by rules_python)"
+    if hermetic_url:
+        interpreter_type = "\"custom\" (pulled from %s)" % hermetic_url
+    print(
+        """
+=============================
+Hermetic Python configuration:
+Version: "{version}"
+Kind: "{py_kind}"
+Interpreter: {interpreter_type}
+Requirements_lock label: "{requirements_lock_label}"
+=====================================
+""".format(
+            version = version,
+            py_kind = py_kind,
+            interpreter_type = interpreter_type,
+            requirements_lock_label = requirements_with_local_wheels,
+        ),
+    )  # buildifier: disable=print
+
+    ctx.file(
+        "py_version.bzl",
+        """
+TF_PYTHON_VERSION = "{version}"
+HERMETIC_PYTHON_VERSION = "{version}"
+HERMETIC_PYTHON_VERSION_KIND = "{py_kind}"
+WHEEL_NAME = "{wheel_name}"
+WHEEL_COLLAB = "{wheel_collab}"
+REQUIREMENTS = "{requirements}"
+REQUIREMENTS_WITH_LOCAL_WHEELS = "{requirements_with_local_wheels}"
+LOCAL_WHEEL_OVERRIDES_LABEL = "{local_wheel_overrides_label}"
+USE_PYWRAP_RULES = {use_pywrap_rules}
+MACOSX_DEPLOYMENT_TARGET = "{macos_deployment_target}"
+HERMETIC_PYTHON_URL = "{hermetic_url}"
+HERMETIC_PYTHON_SHA256 = "{hermetic_sha256}"
+HERMETIC_PYTHON_PREFIX = "{hermetic_prefix}"
+""".format(
+            version = version,
+            py_kind = py_kind,
+            wheel_name = wheel_name,
+            wheel_collab = wheel_collab,
+            requirements = str(requirements),
+            requirements_with_local_wheels = requirements_with_local_wheels,
+            local_wheel_overrides_label = local_wheel_overrides_label,
+            use_pywrap_rules = use_pywrap_rules,
+            macos_deployment_target = macos_deployment_target,
+            hermetic_url = hermetic_url,
+            hermetic_sha256 = hermetic_sha256,
+            hermetic_prefix = hermetic_prefix,
+        ),
+    )
+
+def _get_python_version(ctx):
+    print_warning = False
+
+    version = ctx.os.environ.get("HERMETIC_PYTHON_VERSION", "")
+    if not version:
+        version = ctx.os.environ.get("TF_PYTHON_VERSION", "")
+    if not version:
+        print_warning = True
+        if ctx.attr.default_python_version == "system":
+            python_version_result = ctx.execute(["python3", "--version"])
+            if python_version_result.return_code == 0:
+                version = python_version_result.stdout
+            else:
+                fail("""
+Cannot match hermetic Python version to system Python version.
+System Python was not found.""")
+        else:
+            version = ctx.attr.default_python_version
+
+    version, kind = _parse_python_version(version)
+
+    if print_warning:
+        print("""
+HERMETIC_PYTHON_VERSION variable was not set correctly, using default version.
+Python {} will be used.
+To select Python version, either set HERMETIC_PYTHON_VERSION env variable in
+your shell:
+  export HERMETIC_PYTHON_VERSION=3.12
+OR pass it as an argument to bazel command directly or inside your .bazelrc
+file:
+  --repo_env=HERMETIC_PYTHON_VERSION=3.12
+""".format(version))  # buildifier: disable=print
+    return version, kind
+
+def _parse_python_version(version_str):
+    if version_str.startswith("Python "):
+        py_ver_chunks = version_str[7:].split(".")
+        return "%s.%s" % (py_ver_chunks[0], py_ver_chunks[1]), ""
+    elif "-" in version_str:
+        return version_str.split("-")
+    return version_str, ""
+
+def _rewrite_requirements_with_local_wheels(
+        ctx,
+        py_version,
+        local_wheel_workspaces,
+        base_requirements):
+    """Rewrites the selected lockfile so matching packages resolve from local wheels."""
+
+    os_name = ctx.os.name
+    is_windows = "windows" in os_name.lower()
+    local_file_path_prefix = "file:" if is_windows else "file://"
+
+    # Parse the lockfile once so later helpers can work with package-level data
+    # instead of re-scanning the raw text in multiple places.
+    base_requirement_blocks = _split_requirements_blocks(base_requirements)
+    base_requirement_names = _collect_requirement_names(base_requirement_blocks)
+
+    local_wheels = _collect_local_wheels(
+        ctx,
+        py_version,
+        local_wheel_workspaces,
+        base_requirement_names,
+    )
+
+    merged_requirements_blocks, override_report_entries = _merge_local_wheels_into_base_requirements(
+        base_requirement_blocks,
+        local_wheels,
+        local_file_path_prefix,
+    )
+
+    return merged_requirements_blocks, override_report_entries
+
+def _collect_local_wheels(
+        ctx,
+        py_version,
+        local_wheel_workspaces,
+        base_requirement_names):
+    """Collects the newest matching local wheel for each distribution name."""
+
+    py_ver_marker = "-cp%s-" % py_version.replace(".", "")
+    py_major_ver_marker = "-py%s-" % py_version.split(".")[0]
+    wheels = {}
+
+    if local_wheel_workspaces:
+        for local_wheel_workspace in local_wheel_workspaces:
+            local_wheel_workspace_path = ctx.path(local_wheel_workspace)
+            dist_folder = ctx.attr.local_wheel_dist_folder
+            dist_folder_path = local_wheel_workspace_path.dirname.get_child(dist_folder)
+            if dist_folder_path.exists:
+                dist_wheels = dist_folder_path.readdir()
+                _process_dist_wheels(
+                    dist_wheels,
+                    wheels,
+                    py_ver_marker,
+                    py_major_ver_marker,
+                    ctx.attr.local_wheel_inclusion_list,
+                    ctx.attr.local_wheel_exclusion_list,
+                )
+
+    local_wheels = {}
+    for wheel_name, wheel_path in wheels.items():
+        # Normalize `foo_bar` to `foo-bar`. We assume that, if `foo_bar`
+        # isn't present in requirements, it must be named `foo-bar`. The
+        # exact same distribution name needs to be used to ensure it is
+        # correctly overridden.
+        if "_" in wheel_name and wheel_name not in base_requirement_names:
+            local_package_name = wheel_name.replace("_", "-")
+        else:
+            local_package_name = wheel_name
+
+        local_wheels[_normalize_requirement_name(local_package_name)] = {
+            # Keep the concrete wheel path so the merge step can emit an
+            # explicit `pkg @ file://...whl` direct reference.
+            "wheel_path": wheel_path,
+        }
+
+    return local_wheels
+
+def _merge_local_wheels_into_base_requirements(
+        base_requirement_blocks,
+        local_wheels,
+        local_file_path_prefix):
+    """Rewrites matching requirement blocks to use direct local wheel URLs."""
+
+    # Replace matching lockfile entries with local wheel entries, so they take
+    # precedence in both host-platform and download_only resolution paths.
+    merged_blocks = []
+    override_report_entries = []
+    replaced_packages = {}
+
+    for block in base_requirement_blocks:
+        first_line = block[0] if block else ""
+        package_name = _extract_requirement_name(first_line)
+        if package_name and package_name in local_wheels:
+            local_wheel = local_wheels[package_name]
+            marker = _extract_requirement_marker(first_line)
+
+            comment_lines = []
+            for line in block[1:]:
+                # Keep explanatory comments such as "# via ...", but drop the
+                # old PyPI hash continuations because the replacement now points
+                # pip at a concrete local wheel URL.
+                if line.strip().startswith("#"):
+                    comment_lines.append(line)
+
+            rendered_block, override_report_entry = _make_local_wheel_override(
+                package_name = package_name,
+                local_wheel = local_wheel,
+                marker = marker,
+                comment_lines = comment_lines,
+                local_file_path_prefix = local_file_path_prefix,
+            )
+            merged_blocks.append(rendered_block)
+            override_report_entries.append(override_report_entry)
+            replaced_packages[package_name] = True
+        else:
+            merged_blocks.append("\n".join(block))
+
+    for package_name, local_wheel in local_wheels.items():
+        if package_name not in replaced_packages:
+            # Keep the old behavior of surfacing local wheels even when the base
+            # lockfile has no entry for the package yet.
+            rendered_block, override_report_entry = _make_local_wheel_override(
+                package_name = package_name,
+                local_wheel = local_wheel,
+                marker = "",
+                comment_lines = [],
+                local_file_path_prefix = local_file_path_prefix,
+            )
+            merged_blocks.append(rendered_block)
+            override_report_entries.append(override_report_entry)
+
+    return merged_blocks, override_report_entries
+
+def _print_local_wheel_override_summaries(override_report_entries):
+    if not override_report_entries:
+        return
+
+    print(
+        """
+Local wheel overrides:
+{overrides}
+""".format(
+            overrides = "\n".join([
+                _format_local_wheel_override_summary(override_report_entry)
+                for override_report_entry in override_report_entries
+            ]),
+        ),
+    )  # buildifier: disable=print
+
+def _write_local_wheel_overrides_json(
+        ctx,
+        version,
+        py_kind,
+        requirements_with_local_wheels,
+        override_report_entries):
+    label = "@{repo}//:local_wheel_overrides.json".format(repo = ctx.name)
+    ctx.file(
+        "local_wheel_overrides.json",
+        json.encode_indent({
+            "python_version": version,
+            "python_kind": py_kind,
+            "requirements_lock_label": requirements_with_local_wheels,
+            "overrides": override_report_entries,
+        }),
+    )
+    return label
+
+def _split_requirements_blocks(requirements_content):
+    """Groups a pip-compile-style requirements file into top-level entries."""
+    blocks = []
+    current_block = []
+
+    for line in requirements_content.replace("\r", "").split("\n"):
+        if current_block and not _is_continuation_line(line):
+            blocks.append(current_block)
+            current_block = []
+        current_block.append(line)
+
+    if current_block:
+        blocks.append(current_block)
+
+    return blocks
+
+def _is_continuation_line(line):
+    return line.startswith(" ") or line.startswith("\t")
+
+def _collect_requirement_names(requirement_blocks):
+    """Collects normalized requirement names from lockfile head lines."""
+
+    requirement_names = {}
+    for block in requirement_blocks:
+        first_line = block[0] if block else ""
+        requirement_name = _extract_requirement_name(first_line)
+        if requirement_name:
+            requirement_names[requirement_name] = True
+    return requirement_names
+
+def _extract_requirement_name(line):
+    """Extracts the normalized package name from a requirement head line."""
+
+    stripped_line = line.strip()
+    if not stripped_line or stripped_line.startswith("#") or stripped_line.startswith("-"):
+        return None
+
+    package_name = ""
+    for character in stripped_line.elems():
+        if character.isspace() or character in [">", "<", "~", "=", ";", "[", "@"]:
+            break
+        package_name += character
+
+    if not package_name:
+        return None
+
+    return _normalize_requirement_name(package_name)
+
+def _normalize_requirement_name(package_name):
+    return package_name.replace("_", "-").lower()
+
+def _extract_requirement_marker(line):
+    """Preserves the raw "; ..." condition that controls when a requirement applies.
+
+    For example, this returns 'sys_platform == "linux"' for
+    'jax-cuda12-plugin==0.9.2 ; sys_platform == "linux"'.
+    """
+    stripped_line = line.strip()
+    if stripped_line.endswith("\\"):
+        stripped_line = stripped_line[:-1].rstrip()
+
+    # If the line doesn't contain the separator, partition(";") returns (line, "", "").
+    _, marker_separator, marker = stripped_line.partition(";")
+    if not marker_separator:
+        return ""
+
+    return marker.strip()
+
+def _make_local_wheel_override(
+        package_name,
+        local_wheel,
+        marker,
+        comment_lines,
+        local_file_path_prefix):
+    """Builds both the rewritten requirement block and its reporting metadata."""
+
+    rendered_block = _render_local_override_requirement_block(
+        package_name = package_name,
+        wheel_path = local_wheel["wheel_path"],
+        marker = marker,
+        comment_lines = comment_lines,
+        local_file_path_prefix = local_file_path_prefix,
+    )
+    override_report_entry = _make_local_wheel_override_entry(
+        package_name = package_name,
+        wheel_path = local_wheel["wheel_path"],
+        marker = marker,
+    )
+    return rendered_block, override_report_entry
+
+def _render_local_override_requirement_block(
+        package_name,
+        wheel_path,
+        marker,
+        comment_lines,
+        local_file_path_prefix):
+    requirement_line = "{package_name} @ {local_file_path_prefix}{wheel_path}".format(
+        package_name = package_name,
+        local_file_path_prefix = local_file_path_prefix,
+        wheel_path = wheel_path.realpath,
+    )
+    if marker:
+        requirement_line += " ; " + marker
+
+    rendered_lines = [requirement_line]
+    rendered_lines.extend(comment_lines)
+    return "\n".join(rendered_lines)
+
+def _make_local_wheel_override_entry(package_name, wheel_path, marker):
+    return {
+        "package": package_name,
+        "marker": marker,
+        "wheel": wheel_path.basename,
+    }
+
+def _format_local_wheel_override_summary(override_entry):
+    summary = "  {package_name}".format(package_name = override_entry["package"])
+    if override_entry["marker"]:
+        summary += " ; " + override_entry["marker"]
+    return "{summary} -> {wheel_name}".format(
+        summary = summary,
+        wheel_name = override_entry["wheel"],
+    )
+
+python_repository = repository_rule(
+    implementation = _python_repository_impl,
+    attrs = {
+        "requirements_versions": attr.string_list(
+            mandatory = False,
+            default = [],
+        ),
+        "requirements_locks": attr.label_list(
+            mandatory = False,
+            default = [],
+        ),
+        "local_wheel_workspaces": attr.label_list(
+            mandatory = False,
+            default = [],
+        ),
+        "local_wheel_dist_folder": attr.string(
+            mandatory = False,
+            default = "dist",
+        ),
+        "default_python_version": attr.string(
+            mandatory = False,
+            default = DEFAULT_VERSION,
+        ),
+        "local_wheel_inclusion_list": attr.string_list(
+            mandatory = False,
+            default = ["*"],
+        ),
+        "local_wheel_exclusion_list": attr.string_list(
+            mandatory = False,
+            default = [],
+        ),
+    },
+    environ = [
+        "TF_PYTHON_VERSION",
+        "HERMETIC_PYTHON_VERSION",
+        "HERMETIC_PYTHON_URL",
+        "HERMETIC_PYTHON_SHA256",
+        "HERMETIC_REQUIREMENTS_LOCK",
+        "HERMETIC_PYTHON_PREFIX",
+        "WHEEL_NAME",
+        "WHEEL_COLLAB",
+        "USE_PYWRAP_RULES",
+        "MACOSX_DEPLOYMENT_TARGET",
+    ],
+    local = True,
+)
+
+def _process_dist_wheels(
+        dist_wheels,
+        wheels,
+        py_ver_marker,
+        py_major_ver_marker,
+        local_wheel_inclusion_list,
+        local_wheel_exclusion_list):
+    for wheel in dist_wheels:
+        bn = wheel.basename
+        if not bn.endswith(".whl") or (bn.find(py_ver_marker) < 0 and bn.find(py_major_ver_marker) < 0):
+            continue
+        if not _basic_wildcard_match(bn, local_wheel_inclusion_list, True, False):
+            continue
+        if not _basic_wildcard_match(bn, local_wheel_exclusion_list, False, True):
+            continue
+
+        name_components = bn.split("-")
+        package_name = name_components[0]
+        for name_component in name_components[1:]:
+            if name_component[0].isdigit():
+                break
+            package_name += "-" + name_component
+
+        latest_wheel = wheels.get(package_name, None)
+
+        if not latest_wheel or latest_wheel.basename < wheel.basename:
+            wheels[package_name] = wheel
+
+def _basic_wildcard_match(name, patterns, expected_match_result, match_all):
+    match = False
+    for pattern in patterns:
+        match = False
+        if pattern.startswith("*") and pattern.endswith("*"):
+            match = name.find(pattern[1:-1]) >= 0
+        elif pattern.startswith("*"):
+            match = name.endswith(pattern[1:])
+        elif pattern.endswith("*"):
+            match = name.startswith(pattern[:-1])
+        else:
+            match = name == pattern
+
+        if match_all:
+            if match != expected_match_result:
+                return False
+        elif match == expected_match_result:
+            return True
+
+    return match == expected_match_result
+
+def _custom_python_interpreter_impl(ctx):
+    version = ctx.attr.version
+    version_variant = ctx.attr.version_variant
+    strip_prefix = ctx.attr.strip_prefix.format(
+        version = version,
+        version_variant = version_variant,
+    )
+    urls = [url.format(version = version, version_variant = version_variant) for url in ctx.attr.urls]
+    binary_name = ctx.attr.binary_name
+    if not binary_name:
+        ver_chunks = version.split(".")
+        binary_name = "python%s.%s" % (ver_chunks[0], ver_chunks[1])
+
+    install_dir = "{name}-{version}".format(name = ctx.attr.name, version = version)
+    _exec_and_check(ctx, ["mkdir", install_dir])
+    install_path = ctx.path(install_dir)
+    srcs_dir = "srcs"
+    ctx.download_and_extract(
+        url = urls,
+        stripPrefix = strip_prefix,
+        output = srcs_dir,
+    )
+
+    configure_params = list(ctx.attr.configure_params)
+    if "CC" in ctx.os.environ:
+        configure_params.append("CC={}".format(ctx.os.environ["CC"]))
+    if "CXX" in ctx.os.environ:
+        configure_params.append("CXX={}".format(ctx.os.environ["CXX"]))
+
+    configure_params.append("--prefix=%s" % install_path.realpath)
+    _exec_and_check(
+        ctx,
+        ["./configure"] + configure_params,
+        working_directory = srcs_dir,
+        quiet = False,
+    )
+    res = _exec_and_check(ctx, ["nproc"])
+    cores = 12 if res.return_code != 0 else max(1, int(res.stdout.strip()) - 1)
+    _exec_and_check(ctx, ["make", "-j%s" % cores], working_directory = srcs_dir)
+    _exec_and_check(ctx, ["make", "altinstall"], working_directory = srcs_dir)
+    _exec_and_check(ctx, ["ln", "-s", binary_name, "python3"], working_directory = install_dir + "/bin")
+    tar = "{install_dir}.tgz".format(install_dir = install_dir)
+    _exec_and_check(ctx, ["tar", "czpf", tar, install_dir])
+    _exec_and_check(ctx, ["rm", "-rf", srcs_dir])
+    res = _exec_and_check(ctx, ["sha256sum", tar])
+
+    sha256 = res.stdout.split(" ")[0].strip()
+    tar_path = ctx.path(tar)
+
+    example = """\n\n
+To use newly built Python interpreter add the following code snippet RIGHT AFTER
+python_init_toolchains() in your WORKSPACE file. The code sample should work as
+is but it may need some tuning, if you have special requirements.
+
+```
+load("@rules_python//python:repositories.bzl", "python_register_toolchains")
+python_register_toolchains(
+    name = "python",
+    # By default assume the interpreter is on the local file system, replace
+    # with proper URL if it is not the case.
+    base_url = "file://",
+    ignore_root_user_error = True,
+    python_version = "{version}",
+    tool_versions = {{
+        "{version}": {{
+            # Path to .tar.gz with Python binary. By default it points to .tgz
+            # file in cache where it was built originally; replace with proper
+            # file location, if you moved it somewhere else.
+            "url": "{tar_path}",
+            "sha256": {{
+                # By default we assume Linux x86_64 architecture, eplace with
+                # proper architecture if you were building on a different platform.
+                "x86_64-unknown-linux-gnu": "{sha256}",
+            }},
+            "strip_prefix": "{install_dir}",
+        }},
+    }},
+)
+```
+\n\n""".format(version = version, tar_path = tar_path, sha256 = sha256, install_dir = install_dir)
+
+    instructions = "INSTRUCTIONS-{version}.md".format(version = version)
+    ctx.file(instructions + ".tmpl", example, executable = False)
+    ctx.file(
+        "BUILD.bazel",
+        """
+genrule(
+    name = "{name}",
+    srcs = ["{tar}", "{instructions}.tmpl"],
+    outs = ["{install_dir}.tar.gz", "{instructions}"],
+    cmd = "cp $(location {tar}) $(location {install_dir}.tar.gz); cp $(location {instructions}.tmpl) $(location {instructions})",
+    visibility = ["//visibility:public"],
+)
+     """.format(
+            name = ctx.attr.name,
+            tar = tar,
+            install_dir = install_dir,
+            instructions = instructions,
+        ),
+        executable = False,
+    )
+
+    print(example)  # buildifier: disable=print
+
+custom_python_interpreter = repository_rule(
+    implementation = _custom_python_interpreter_impl,
+    attrs = {
+        "urls": attr.string_list(),
+        "strip_prefix": attr.string(),
+        "binary_name": attr.string(mandatory = False),
+        "version": attr.string(),
+        "version_variant": attr.string(),
+        "configure_params": attr.string_list(
+            mandatory = False,
+            default = ["--enable-optimizations"],
+        ),
+    },
+)
+
+def _exec_and_check(ctx, command, fail_on_error = True, quiet = False, **kwargs):
+    res = ctx.execute(command, quiet = quiet, **kwargs)
+    if fail_on_error and res.return_code != 0:
+        fail("""
+Failed to execute command: `{command}`
+Exit Code: {code}
+STDERR: {stderr}
+        """.format(
+            command = command,
+            code = res.return_code,
+            stderr = res.stderr,
+        ))
+    return res

third_party/py/rules_cc_protobuf.patch

@@ -0,0 +1,29 @@
+--- a/cc/defs.bzl
++++ b/cc/defs.bzl
+@@ -18,3 +18,4 @@
+ """Starlark rules for building C++ projects."""
+
++load("@com_google_protobuf//bazel:cc_proto_library.bzl", _protobuf_cc_proto_library = "cc_proto_library")
+ load("//cc:cc_binary.bzl", _cc_binary = "cc_binary")
+@@ -48,20 +49,3 @@
+ objc_import = _objc_import
+
+-# DEPRECATED: use rule from com_google_protobuf repository
+-def cc_proto_library(**kwargs):
+-    """Deprecated redirection macro for cc_proto_library.
+-
+-    Use cc_proto_library from com_google_protobuf.
+-
+-    On Bazel <8, redirects to native.cc_proto_library.
+-    On Bazel >=8, redirects to a mock rule, that fails when analyzed.
+-    This allows for a gradual migration away from this macro.
+-
+-    Args:
+-      **kwargs: passed directly into cc_proto_library
+-    """
+-    __cc_proto_library = getattr(native, "cc_proto_library", _cc_proto_library)
+-    if "deprecation" not in kwargs:
+-        __cc_proto_library(deprecation = CC_PROTO_LIBRARY_DEPRECATION, **kwargs)
+-    else:
+-        __cc_proto_library(**kwargs)
++cc_proto_library = _protobuf_cc_proto_library

third_party/py/rules_python_freethreaded.patch

@@ -0,0 +1,45 @@
+diff --git a/python/private/python_register_toolchains.bzl b/python/private/python_register_toolchains.bzl
+index 9e75c419..ba6d8f7a 100644
+--- a/python/private/python_register_toolchains.bzl
++++ b/python/private/python_register_toolchains.bzl
+@@ -94,6 +94,7 @@ def python_register_toolchains(
+     minor_mapping = minor_mapping or MINOR_MAPPING
+ 
+     python_version = full_version(version = python_version, minor_mapping = minor_mapping)
++    python_version_kind = kwargs.pop("python_version_kind", "")
+ 
+     toolchain_repo_name = "{name}_toolchains".format(name = name)
+ 
+@@ -174,6 +175,7 @@ def python_register_toolchains(
+         name = name + "_host",
+         platforms = loaded_platforms,
+         python_version = python_version,
++        python_version_kind = python_version_kind,
+     )
+ 
+     toolchains_repo(
+diff --git a/python/private/toolchains_repo.bzl b/python/private/toolchains_repo.bzl
+index f7ff19c3..458f2885 100644
+--- a/python/private/toolchains_repo.bzl
++++ b/python/private/toolchains_repo.bzl
+@@ -445,6 +445,10 @@ Full python version, Major.Minor.Micro.
+ Only set in workspace calls.
+ """,
+         ),
++        "python_version_kind": attr.string(
++            doc = "Python version kind, e.g. ft (free-threaded)",
++            default = ""
++        ),
+         "python_versions": attr.string_dict(
+             doc = """
+ If set, the Python version for the corresponding selected platform. Values in
+@@ -603,6 +607,9 @@ def _get_host_impl_repo_name(*, rctx, logger, python_version, os_name, cpu_name,
+         else:
+             candidates = [preference]
+ 
++        if rctx.attr.python_version_kind == "ft":
++            candidates = [c for c in candidates if c[0].endswith("freethreaded")]
++
+     if candidates:
+         platform_name, meta = candidates[0]
+         suffix = meta.impl_repo_name

third_party/py/rules_python_pip_version.patch

@@ -0,0 +1,26 @@
+diff --git a/python/private/pypi/deps.bzl b/python/private/pypi/deps.bzl
+index 8949ed4a..8d0ab0e7 100644
+--- a/python/private/pypi/deps.bzl
++++ b/python/private/pypi/deps.bzl
+@@ -51,8 +51,8 @@ _RULE_DEPS = [
+     ),
+     (
+         "pypi__packaging",
+-        "https://files.pythonhosted.org/packages/49/df/1fceb2f8900f8639e278b056416d49134fb8d84c5942ffaa01ad34782422/packaging-24.0-py3-none-any.whl",
+-        "2ddfb553fdf02fb784c234c7ba6ccc288296ceabec964ad2eae3777778130bc5",
++        "https://files.pythonhosted.org/packages/88/ef/eb23f262cca3c0c4eb7ab1933c3b1f03d021f2c48f54763065b6f0e321be/packaging-24.2-py3-none-any.whl",
++        "09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759",
+     ),
+     (
+         "pypi__pep517",
+@@ -61,8 +61,8 @@ _RULE_DEPS = [
+     ),
+     (
+         "pypi__pip",
+-        "https://files.pythonhosted.org/packages/8a/6a/19e9fe04fca059ccf770861c7d5721ab4c2aebc539889e97c7977528a53b/pip-24.0-py3-none-any.whl",
+-        "ba0d021a166865d2265246961bec0152ff124de910c5cc39f1156ce3fa7c69dc",
++        "https://files.pythonhosted.org/packages/ef/7d/500c9ad20238fcfcb4cb9243eede163594d7020ce87bd9610c9e02771876/pip-24.3.1-py3-none-any.whl",
++        "3790624780082365f47549d032f3770eeb2b1e8bd1f7b2e02dace1afa361b4ed",
+     ),
+     (
+         "pypi__pip_tools",

third_party/py/rules_python_scope.patch

@@ -0,0 +1,33 @@
+diff --git a/python/config_settings/BUILD.bazel b/python/config_settings/BUILD.bazel
+index 369989eb..b20a0da6 100644
+--- a/python/config_settings/BUILD.bazel
++++ b/python/config_settings/BUILD.bazel
+@@ -245,7 +245,6 @@ label_flag(
+ bool_flag(
+     name = "experimental_python_import_all_repositories",
+     build_setting_default = True,
+-    scope = "universal",
+     visibility = ["//visibility:public"],
+ )
+
+@@ -253,20 +252,17 @@ bool_flag(
+     name = "build_python_zip",
+     build_setting_default = config.build_python_zip_default,
+     help = "Build python executable zip. Defaults to on on Windows, off on other platforms",
+-    scope = "universal",
+     visibility = ["//visibility:public"],
+ )
+
+ bool_flag(
+     name = "incompatible_default_to_explicit_init_py",
+     build_setting_default = False,
+-    scope = "universal",
+     visibility = ["//visibility:public"],
+ )
+
+ string_flag(
+     name = "python_path",
+     build_setting_default = "python",
+-    scope = "universal",
+     visibility = ["//visibility:public"],
+ )

third_party/py/rules_python_versions.patch

@@ -0,0 +1,117 @@
+diff --git a/python/versions.bzl b/python/versions.bzl
+index 3299a9f1..27ec699f 100644
+--- a/python/versions.bzl
++++ b/python/versions.bzl
+@@ -992,6 +992,71 @@ TOOL_VERSIONS = {
+             "x86_64-unknown-linux-gnu-freethreaded": "python/install",
+         },
+     },
++    "3.14.6": {
++        "url": {
++            "aarch64-apple-darwin": "20260610/cpython-3.14.6+20260610-aarch64-apple-darwin-install_only.tar.gz",
++            "aarch64-apple-darwin-freethreaded": "20260610/cpython-3.14.6+20260610-aarch64-apple-darwin-freethreaded-install_only.tar.gz",
++            "aarch64-pc-windows-msvc": "20260610/cpython-3.14.6+20260610-aarch64-pc-windows-msvc-install_only.tar.gz",
++            "aarch64-pc-windows-msvc-freethreaded": "20260610/cpython-3.14.6+20260610-aarch64-pc-windows-msvc-freethreaded-install_only.tar.gz",
++            "aarch64-unknown-linux-gnu": "20260610/cpython-3.14.6+20260610-aarch64-unknown-linux-gnu-install_only.tar.gz",
++            "aarch64-unknown-linux-gnu-freethreaded": "20260610/cpython-3.14.6+20260610-aarch64-unknown-linux-gnu-freethreaded-install_only.tar.gz",
++            "ppc64le-unknown-linux-gnu": "20260610/cpython-3.14.6+20260610-ppc64le-unknown-linux-gnu-install_only.tar.gz",
++            "ppc64le-unknown-linux-gnu-freethreaded": "20260610/cpython-3.14.6+20260610-ppc64le-unknown-linux-gnu-freethreaded-install_only.tar.gz",
++            "riscv64-unknown-linux-gnu": "20260610/cpython-3.14.6+20260610-riscv64-unknown-linux-gnu-install_only.tar.gz",
++            "riscv64-unknown-linux-gnu-freethreaded": "20260610/cpython-3.14.6+20260610-riscv64-unknown-linux-gnu-freethreaded-install_only.tar.gz",
++            "s390x-unknown-linux-gnu": "20260610/cpython-3.14.6+20260610-s390x-unknown-linux-gnu-install_only.tar.gz",
++            "s390x-unknown-linux-gnu-freethreaded": "20260610/cpython-3.14.6+20260610-s390x-unknown-linux-gnu-freethreaded-install_only.tar.gz",
++            "x86_64-apple-darwin": "20260610/cpython-3.14.6+20260610-x86_64-apple-darwin-install_only.tar.gz",
++            "x86_64-apple-darwin-freethreaded": "20260610/cpython-3.14.6+20260610-x86_64-apple-darwin-freethreaded-install_only.tar.gz",
++            "x86_64-pc-windows-msvc": "20260610/cpython-3.14.6+20260610-x86_64-pc-windows-msvc-install_only.tar.gz",
++            "x86_64-pc-windows-msvc-freethreaded": "20260610/cpython-3.14.6+20260610-x86_64-pc-windows-msvc-freethreaded-install_only.tar.gz",
++            "x86_64-unknown-linux-gnu": "20260610/cpython-3.14.6+20260610-x86_64-unknown-linux-gnu-install_only.tar.gz",
++            "x86_64-unknown-linux-gnu-freethreaded": "20260610/cpython-3.14.6+20260610-x86_64-unknown-linux-gnu-freethreaded-install_only.tar.gz",
++            "x86_64-unknown-linux-musl": "20260610/cpython-3.14.6+20260610-x86_64-unknown-linux-musl-install_only.tar.gz",
++        },
++        "sha256": {
++            "aarch64-apple-darwin": "953db72ff2dea68b5112231b1ba77163ec9114f87c7ece530b3ea742a3b492c5",
++            "aarch64-apple-darwin-freethreaded": "c4b8be603b2b08cc5e9ef8905b02f646fe1f7e30b40270a54e526b18aa2e0e21",
++            "aarch64-pc-windows-msvc": "48e5b0e58c866ce9fd01886fc1ab4b5293160b9dce2a7841cae7e7d894824f1a",
++            "aarch64-pc-windows-msvc-freethreaded": "813dfad8c8ef72d1fbbd2797cbe27aa6d15efe8a8268d6c9592310a504a7d3d2",
++            "aarch64-unknown-linux-gnu": "9bc4a8b40e849c927158403552ffea25627f17776e5200156b131ac78b049384",
++            "aarch64-unknown-linux-gnu-freethreaded": "e0e7d3f82c2a8eee56846e25e58da6b62062e4d3254fa612e09bb66bc5edc68e",
++            "ppc64le-unknown-linux-gnu": "b2ee623296f19cdeab78fecfc9271e499d6b0384f8106c3b23ce0edb45157b28",
++            "ppc64le-unknown-linux-gnu-freethreaded": "39ed498c6fcca3a69b88aaf0f3cca5fad79c654872c2f509200a995bc663b6f4",
++            "riscv64-unknown-linux-gnu": "9504a5c4137b9f0397ba6b83166536e9e007d46dade82c57e7357faaed809d95",
++            "riscv64-unknown-linux-gnu-freethreaded": "18bfc14392fe0a0dd8f621a188f1b6d3bdda891cfe403327fc10566c8802b79b",
++            "s390x-unknown-linux-gnu": "69c2edceeecca2afadb2ea06ad07b7a7bfe648c3a2837b377f8a7aefde1fc2d6",
++            "s390x-unknown-linux-gnu-freethreaded": "ea2032e2ed1fc3758db904ab616e48e9955253485a6ec01b3c18381413845a95",
++            "x86_64-apple-darwin": "43f9833eabca62f2c9d8d3a3fb35835d6116386d871cb842c1ec2f08a8d27bd7",
++            "x86_64-apple-darwin-freethreaded": "b9e4794d034cb1e3c172a1c58a50a505e7c3682cb4d73fb6316c57e71ca669cc",
++            "x86_64-pc-windows-msvc": "8bb7fa0570d77cfb321df1763619f2cfaadbe1502c7717773cbe023ebd639472",
++            "x86_64-pc-windows-msvc-freethreaded": "c50391acda90e570d6ffffef8cf098ea69e2a5c605f677822991cc2628e25fc5",
++            "x86_64-unknown-linux-gnu": "288f0e2c728f2bec803c4bffbd0ad3940190221ef39c1540b69d5e4ad6861b10",
++            "x86_64-unknown-linux-gnu-freethreaded": "cd60c024956a06318cba8d136ca6f2861e11123a5baadff76c822fa3ce4780c5",
++            "x86_64-unknown-linux-musl": "c55940c8ef8cfa73a8a5bc2c1cf3ea37cc87c92f7bd208adbbd283f4a1df652b",
++        },
++        "strip_prefix": {
++            "aarch64-apple-darwin": "python",
++            "aarch64-apple-darwin-freethreaded": "python",
++            "aarch64-pc-windows-msvc": "python",
++            "aarch64-pc-windows-msvc-freethreaded": "python",
++            "aarch64-unknown-linux-gnu": "python",
++            "aarch64-unknown-linux-gnu-freethreaded": "python",
++            "ppc64le-unknown-linux-gnu": "python",
++            "ppc64le-unknown-linux-gnu-freethreaded": "python",
++            "riscv64-unknown-linux-gnu": "python",
++            "riscv64-unknown-linux-gnu-freethreaded": "python",
++            "s390x-unknown-linux-gnu": "python",
++            "s390x-unknown-linux-gnu-freethreaded": "python",
++            "x86_64-apple-darwin": "python",
++            "x86_64-apple-darwin-freethreaded": "python",
++            "x86_64-pc-windows-msvc": "python",
++            "x86_64-pc-windows-msvc-freethreaded": "python",
++            "x86_64-unknown-linux-gnu": "python",
++            "x86_64-unknown-linux-gnu-freethreaded": "python",
++            "x86_64-unknown-linux-musl": "python",
++        },
++    },
+     "3.15.0a1": {
+         "url": "20251031/cpython-{python_version}+20251031-{platform}-{build}.{ext}",
+         "sha256": {
+@@ -1091,7 +1156,7 @@ MINOR_MAPPING = {
+     "3.11": "3.11.14",
+     "3.12": "3.12.12",
+     "3.13": "3.13.11",
+-    "3.14": "3.14.2",
++    "3.14": "3.14.6",
+     "3.15": "3.15.0a2",
+ }
+
+@@ -1272,20 +1337,13 @@ def get_release_info(platform, python_version, base_url = DEFAULT_RELEASE_BASE_U
+     for u in url:
+         p, _, _ = platform.partition(FREETHREADED)
+ 
+-        # Assume an unknown release_id is a newer url format
+-        release_id = 99999999
+-        url_parts = u.split("/")
+-        if len(url_parts) >= 2 and url_parts[-2].isdigit():
+-            maybe_release_id = url_parts[-2]
+-            release_id = int(maybe_release_id)
+-
+         if FREETHREADED.lstrip("-") in platform:
+             build = "{}+{}-full".format(
+                 FREETHREADED.lstrip("-"),
+                 {
+                     "aarch64-apple-darwin": "pgo+lto",
+                     "aarch64-pc-windows-msvc": "pgo",
+-                    "aarch64-unknown-linux-gnu": "lto" if release_id < 20250702 else "pgo+lto",
++                    "aarch64-unknown-linux-gnu": "pgo+lto",
+                     "ppc64le-unknown-linux-gnu": "lto",
+                     "riscv64-unknown-linux-gnu": "lto",
+                     "s390x-unknown-linux-gnu": "lto",
+@@ -1298,9 +1356,6 @@ def get_release_info(platform, python_version, base_url = DEFAULT_RELEASE_BASE_U
+         else:
+             build = INSTALL_ONLY
+ 
+-        if WINDOWS_NAME in platform and release_id < 20250317:
+-            build = "shared-" + build
+-
+         release_filename = u.format(
+             platform = p,
+             python_version = python_version,