From e3e021fd151d23ffedafe8a78f7150f97a8a33ac Mon Sep 17 00:00:00 2001
From: Ruturaj4 <Ruturaj.vaidya@amd.com>
Date: Tue, 10 Feb 2026 08:45:48 -0600
Subject: [PATCH] Bump up jax and xla versions

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

diff --git a/jax_rocm_plugin/third_party/jax/workspace.bzl b/jax_rocm_plugin/third_party/jax/workspace.bzl
index ccc5352339..10479b6238 100644
--- a/jax_rocm_plugin/third_party/jax/workspace.bzl
+++ b/jax_rocm_plugin/third_party/jax/workspace.bzl
@@ -1,7 +1,7 @@
 load("//third_party:repo.bzl", "amd_http_archive")
 
-COMMIT = "fbfa695aea59ed578b81d8fc72ab23bba5d2cfaa"
-SHA = "b740b326b468ce7ef967fbfab0accfb19850fab9c43ab6a3a37112eff34223c2"
+COMMIT = "21d5e0bf8f07fb7e455f627284e490e6e3068176"
+SHA = "a05b6d06ef0609b7a9ad4ea7195dca564e17d99057d589edc9cc8c487d3098de"
 
 def repo():
     amd_http_archive(
diff --git a/jax_rocm_plugin/third_party/xla/workspace.bzl b/jax_rocm_plugin/third_party/xla/workspace.bzl
index 10645fe5b6..9c09e621af 100644
--- a/jax_rocm_plugin/third_party/xla/workspace.bzl
+++ b/jax_rocm_plugin/third_party/xla/workspace.bzl
@@ -16,8 +16,8 @@ load("//third_party:repo.bzl", "amd_http_archive")
 #    curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
 #    and update XLA_SHA256 with the result.
 
-XLA_COMMIT = "24c5f10ae8fc24aefd20b43c501ade7f66fd0cfd"
-XLA_SHA256 = "f00db8761e86bcb51b52e64bc983717181050c8752c040e33ecb1429d861c30b"
+XLA_COMMIT = "107e89fa589603c5c4a0c60ff333130f0ee1b5c8"
+XLA_SHA256 = "1ab102ec37151f8e70cef456a071c28f0a188953ab1bfbb8d9e357edaf6ab9e3"
 
 def repo():
     amd_http_archive(