diff --git a/third_party/jax/0.11.1/README.md b/third_party/jax/0.11.1/README.md
index 11ff39ede..5eb6b782b 100644
--- a/third_party/jax/0.11.1/README.md
+++ b/third_party/jax/0.11.1/README.md
@@ -1,10 +1,10 @@
 # JAX 0.11.1
 
 - JAX commit: `2d66622450e2c8633cda2307688ef7aa294bd6eb`
-- XLA commit: `f85cfbe2e260907a52bbbe85942462377c8fdb62`
-- rules_ml_toolchain: `73cb731fed3c9215033c5e21e64906f376cf47e8`
+- XLA commit: `dcf304bc5dca1932b99f740b911dbd73631a1a69`
+- rules_ml_toolchain: `73cb731fed3ccf7551beac710bf1c5dbeb8be298`
 - abseil-cpp: `20260526.0`
-- libtpu: `0.0.48`
+- libtpu: `0.0.46.1`
 - `RAIDEN_JAX`: `1101`
 
 ## Patches
diff --git a/third_party/jax/0.11.1/deps.bzl b/third_party/jax/0.11.1/deps.bzl
index f5104fa13..12f2fe120 100644
--- a/third_party/jax/0.11.1/deps.bzl
+++ b/third_party/jax/0.11.1/deps.bzl
@@ -24,21 +24,21 @@ DEPS = {
         "third_party/jax/0.11.1/patches/py/jax_remove_local_wheels.patch",
     ],
     "xla_repo": "https://github.com/openxla/xla",
-    "xla_commit": "f85cfbe2e260907a52bbbe85942462377c8fdb62",
-    "xla_integrity": "sha256-6g6q14v0kXvj0l2iE7k/Q7v3N8b5N3a9u4o0m6Y2o3g=",
+    "xla_commit": "dcf304bc5dca1932b99f740b911dbd73631a1a69",
+    "xla_integrity": "sha256-TIns//WmYqbt+04tQD/t9VvkCjoAeePC+LpHs3wW6qs=",
     "xla_patches": [
         "third_party/xla/future_sfinae.patch",
         "third_party/xla/attribute_map_static_assert.patch",
         "third_party/xla/attribute_map_cc_static_assert.patch",
     ],
     "rules_ml_toolchain_repo": "https://github.com/google-ml-infra/rules_ml_toolchain",
-    "rules_ml_toolchain_commit": "73cb731fed3c9215033c5e21e64906f376cf47e8",
-    "rules_ml_toolchain_integrity": "sha256-oD1n5+UfPq1d8T9sK0m2W4j6N8b5N3a9u4o0m6Y2o3g=",
+    "rules_ml_toolchain_commit": "73cb731fed3ccf7551beac710bf1c5dbeb8be298",
+    "rules_ml_toolchain_integrity": "sha256-dcy3xNxpk0PwLQLwtbqm2sCcAT1GuStg8YVjRI6lnnc=",
     "rules_ml_toolchain_patches": [
         "third_party/jax/0.11.1/patches/rules_ml_toolchain/no_register_toolchains.patch",
     ],
     "absl_repo": "https://github.com/abseil/abseil-cpp",
     "absl_version": "20260526.0",
     "absl_patches": [],
-    "libtpu_version": "0.0.48",
+    "libtpu_version": "0.0.46.1",
 }
diff --git a/third_party/jax/0.11.1/patches/py/jax_remove_local_wheels.patch b/third_party/jax/0.11.1/patches/py/jax_remove_local_wheels.patch
index 05cd54bd0..e1bdb80b2 100644
--- a/third_party/jax/0.11.1/patches/py/jax_remove_local_wheels.patch
+++ b/third_party/jax/0.11.1/patches/py/jax_remove_local_wheels.patch
@@ -1,6 +1,6 @@
 --- a/MODULE.bazel
 +++ b/MODULE.bazel
-@@ -145,19 +145,7 @@
+@@ -145,7 +145,7 @@
  python.toolchain(python_version = "3.15")
  
  ### Pypi
@@ -9,8 +9,7 @@
  
  [pip.parse(
      download_only = True,
-     experimental_index_url = "https://us-python.pkg.dev/ml-oss-artifacts-published/pypi-mirror/simple",
-     extra_hub_aliases = {
+@@ -154,18 +154,6 @@
          "numpy": ["numpy_headers"],
      },
      hub_name = "jax_pypi",
@@ -27,3 +26,5 @@
 -        "scipy": "dist/scipy-*.whl",
 -    },
      python_version = python_version,
+     requirements_by_platform = {} if python_version in [
+         "3.12",