diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp
index f92b54a2af..efa1a96db9 100644
--- a/deps/ReactantExtra/API.cpp
+++ b/deps/ReactantExtra/API.cpp
@@ -167,7 +167,8 @@
 #include "xla/stream_executor/device_description.h"
 #endif
 
-#include "jaxlib/mosaic/dialect/tpu/tpu_dialect.h"
+// Broken upstream x/ref https://github.com/jax-ml/jax/issues/33344
+// #include "jaxlib/mosaic/dialect/tpu/tpu_dialect.h"
 
 #include "llvm/Support/ExtensibleRTTI.h"
 #include <llvm/Support/FileSystem.h>
diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD
index 7254ceb3bc..eaee062831 100644
--- a/deps/ReactantExtra/BUILD
+++ b/deps/ReactantExtra/BUILD
@@ -1175,7 +1175,9 @@ cc_library(
         "@com_google_absl//absl/log:globals",
         "@llvm-project//mlir:CAPIIRObjects",
         "@llvm-project//mlir:CAPILLVMObjects",
-        "@jax//jaxlib/mosaic:tpu_dialect_capi_objects",
+        
+        # Broken upstream x/ref https://github.com/jax-ml/jax/issues/33344
+        # "@jax//jaxlib/mosaic:tpu_dialect_capi_objects",
         "@jax//jaxlib/triton:triton_dialect_capi_objects",
         "@xla//xla/pjrt/plugin/xla_cpu:xla_cpu_pjrt_client",
         "@stablehlo//:linalg_passes",