Update libtpu installation index path

Update the libtpu installation index to https://storage.googleapis.com/libtpu-wheels/index.html, which includes both stable and nightly libtpu versions, as per the cloud libtpu team's guidance.

Also update the libtpu version in the setup.py. It starts to differ from the TF version to support JAX, and it requires manual updates for new releases for now.

PiperOrigin-RevId: 728741103

Author: kanglant

ci/official/envs/linux_x86_tpu

@@ -20,4 +20,4 @@ TFCI_LIB_SUFFIX="-tpu-linux-x86_64"
 TFCI_WHL_BAZEL_TEST_ENABLE=0
 TFCI_WHL_IMPORT_TEST_ENABLE=0
 TFCI_WHL_SIZE_LIMIT=580M
-TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS="-f https://storage.googleapis.com/libtpu-tf-releases/index.html"
+TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS="-f https://storage.googleapis.com/libtpu-wheels/index.html"

tensorflow/tools/pip_package/setup.py

@@ -50,6 +50,9 @@
 # tensorflow/core/public/version.h
 _VERSION = '2.19.0-rc0'
 
+# Update this version when a new libtpu stable version is released.
+LATEST_RELEASE_LIBTPU_VERSION = '0.0.9'
+NEXT_LIBTPU_VERSION = '0.0.10'
 
 # We use the same setup.py for all tensorflow_* packages and for the nightly
 # equivalents (tf_nightly_*). The package is controlled from the argument line
@@ -310,17 +313,14 @@ def find_files(pattern, root):
   # timing of these tests, the UTC date from eight hours ago is expected to be a
   # valid version.
   _libtpu_version = standard_or_nightly(
-      _VERSION.replace('-', ''),
-      '0.1.dev'
+      LATEST_RELEASE_LIBTPU_VERSION,
+      NEXT_LIBTPU_VERSION + '.dev'
       + (
           datetime.datetime.now(tz=datetime.timezone.utc)
           - datetime.timedelta(hours=8)
-      ).strftime('%Y%m%d'),
+      ).strftime('%Y%m%d') + '+nightly',
   )
-  if _libtpu_version.startswith('0.1'):
-    REQUIRED_PACKAGES.append([f'libtpu-nightly=={_libtpu_version}'])
-  else:
-    REQUIRED_PACKAGES.append([f'libtpu=={_libtpu_version}'])
+  REQUIRED_PACKAGES.append([f'libtpu=={_libtpu_version}'])
   CONSOLE_SCRIPTS.extend([
       'start_grpc_tpu_worker = tensorflow.python.tools.grpc_tpu_worker:run',
       ('start_grpc_tpu_service = '