Enable profiler_test for TPU's

cliveverghese · Google-ML-Automation · commit e9925ee03d5b · 2025-06-02T12:06:14.000-07:00
PiperOrigin-RevId: 766289665
diff --git a/tests/BUILD b/tests/BUILD
@@ -1136,6 +1136,7 @@ jax_multiplatform_test(
     enable_backends = [
         "cpu",
         "gpu",
+        "tpu",
     ],
     deps = [
         "//jax:profiler",
