From d477b921215d5030d09020a70df7f5cd46bb62dd Mon Sep 17 00:00:00 2001
From: Yash Katariya <yashkatariya@google.com>
Date: Mon, 18 Sep 2023 11:02:31 -0700
Subject: [PATCH] Start jax and jaxlib 0.4.16 release

PiperOrigin-RevId: 566354738
---
 setup.py                      | 6 +++---
 third_party/xla/workspace.bzl | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index c2929c66dd5a..f4fdb7993672 100644
--- a/setup.py
+++ b/setup.py
@@ -22,13 +22,13 @@
 
 project_name = 'jax'
 
-_current_jaxlib_version = '0.4.15'
+_current_jaxlib_version = '0.4.16'
 # The following should be updated with each new jaxlib release.
-_latest_jaxlib_version_on_pypi = '0.4.15'
+_latest_jaxlib_version_on_pypi = '0.4.14'
 _available_cuda11_cudnn_versions = ['86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '89'
-_libtpu_version = '0.1.dev20230830'
+_libtpu_version = '0.1.dev20230918'
 
 def load_version_module(pkg_path):
   spec = importlib.util.spec_from_file_location(
diff --git a/third_party/xla/workspace.bzl b/third_party/xla/workspace.bzl
index d87ed7ca4014..4d0f8fc287fc 100644
--- a/third_party/xla/workspace.bzl
+++ b/third_party/xla/workspace.bzl
@@ -20,8 +20,8 @@ load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 #    curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
 #    and update XLA_SHA256 with the result.
 
-XLA_COMMIT = "70ae45e23082c7569c381648997bbeecd1c7b281"
-XLA_SHA256 = "7780680d34d886dd3cee414b23b4d026b8f1057ae756cd3ac6c0990ff9a0cd78"
+XLA_COMMIT = "f089a9224682e9658b6e20f7350af4909464378d"
+XLA_SHA256 = "5fc8bb5da7f3cda7f5be4589f8be744464281b949d2902e250c1a39580eaabc7"
 
 def repo():
     tf_http_archive(