diff --git a/third_party/xla/workspace.bzl b/third_party/xla/workspace.bzl
index 7745d40ae..d7feaaeaf 100644
--- a/third_party/xla/workspace.bzl
+++ b/third_party/xla/workspace.bzl
@@ -1,10 +1,8 @@
 """Loads XLA."""
 
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
-# load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
+load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
 load("//:workspace.bzl", "XLA_PATCHES")
-XLA_COMMIT = "1bb4fc18e73faa1c001d96bfe3a22f733987b018"
-XLA_SHA256 = ""
 
 def repo():
     http_archive(
diff --git a/workspace.bzl b/workspace.bzl
index 6618adce4..b3e480934 100644
--- a/workspace.bzl
+++ b/workspace.bzl
@@ -1,4 +1,4 @@
-JAX_COMMIT = "afcb21ddf1075c908a040f23be07f083eb58f473"
+JAX_COMMIT = "41993fdb24021473f6e2db8fdeeeb8b6d3f52630"
 JAX_SHA256 = ""
 
 ENZYME_COMMIT = "565163635fe98c7e0da004b4ac6dd2cd45bc880c"