From b9c602b8d7227ad30ac4310846150bef22943e02 Mon Sep 17 00:00:00 2001
From: Skye Wanderman-Milne <skyewm@google.com>
Date: Tue, 3 Oct 2023 13:20:25 -0700
Subject: [PATCH] Update XLA commit and versions for jax 0.4.17 release

---
 setup.py                      | 4 ++--
 third_party/xla/workspace.bzl | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index c02eff6e7f6d..ccf050fbbd69 100644
--- a/setup.py
+++ b/setup.py
@@ -22,13 +22,13 @@
 
 project_name = 'jax'
 
-_current_jaxlib_version = '0.4.16'
+_current_jaxlib_version = '0.4.17'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.16'
 _available_cuda11_cudnn_versions = ['86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '89'
-_libtpu_version = '0.1.dev20230918'
+_libtpu_version = '0.1.dev20231003'
 
 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 a1f74f3104ab..517b7f7a5d20 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 = "7a19856d74569fd1f765cd03bdee84e3b1fdc579"
-XLA_SHA256 = "2b632f8adb63674ce2b8b48a84729a35226af8cce68431a7804fec8ca6eac61b"
+XLA_COMMIT = "7a32bc798bfd881ccf8fef4a0e5284240f233808"
+XLA_SHA256 = "ff75887de4107928e5c2688323c1162795fd211ab4e4fed77d3d505447bf9403"
 
 def repo():
     tf_http_archive(