diff --git a/tests/BUILD b/tests/BUILD
index b5a99b254c16..07069b08e7b2 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -692,6 +692,9 @@ jax_test(
         "gpu": [
             "noasan",  # Times out under asan.
         ],
+        "tpu": [
+            "noasan",  # Times out under asan.
+        ],
     },
     shard_count = {
         "cpu": 10,