Add TODO to run TPU interpret mode tests in parallel.

PiperOrigin-RevId: 762143977

Author: jburnim

tests/pallas/tpu_pallas_interpret_distributed_test.py

@@ -38,6 +38,8 @@
 P = jax.sharding.PartitionSpec
 
 
+# TODO(jburnim): Figure out how to safely run different instance of TPU
+# interpret mode in parallel, and then remove this decorator.
 @jtu.thread_unsafe_test_class()
 class InterpretDistributedTest(jtu.JaxTestCase):
   def setUp(self):

tests/pallas/tpu_pallas_interpret_test.py

@@ -82,6 +82,8 @@ def grid_points(self):
     return self._grid_points
 
 
+# TODO(jburnim): Figure out how to safely run different instance of TPU
+# interpret mode in parallel, and then remove this decorator.
 @jtu.thread_unsafe_test_class()
 class InterpretTest(jtu.JaxTestCase):