[jax2tf] Refine the disabling of jax2tf_test, for versions <= 2.19.1

Previously we disabled the jax2tf_test for older versions of TF.
Re-enable for 2.19.1 and higher.

Author: gnecula

jax/experimental/jax2tf/tests/jax2tf_test.py

@@ -48,11 +48,17 @@
 config.parse_flags_with_absl()
 
 
-@unittest.skip("Failing after jax 0.6.1 release")
 class Jax2TfTest(tf_test_util.JaxToTfTestCase):
 
   def setUp(self):
     super().setUp()
+    versions = tf.version.VERSION.split(".")
+    if versions < ["2", "19", "1"]:
+      # StableHLO changed on March 18th, 2025 ,to version 1.10.0, and this
+      # introduces ops like vhlo_sine_v2. These ops require a TF version
+      # released after this date.
+      self.skipTest("Need version of TensorFlow at least 2.19.1")
+
     # One TF device of each device_type
     self.tf_devices = []
     for tf_device in (tf.config.list_logical_devices("TPU") +
@@ -1783,11 +1789,17 @@ def func():
     jax_result = func()
     self.assertEqual(tf_result, jax_result)
 
-@unittest.skip("Failing after jax 0.6.1 release")
+
 class Jax2TfVersioningTest(tf_test_util.JaxToTfTestCase):
   # Use a separate test case with the default jax_serialization_version
   def setUp(self):
     self.use_max_serialization_version = False
+    versions = tf.version.VERSION.split(".")
+    if versions < ["2", "19", "1"]:
+      # StableHLO changed on March 18th, 2025 ,to version 1.10.0, and this
+      # introduces ops like vhlo_sine_v2. These ops require a TF version
+      # released after this date.
+      self.skipTest("Need version of TensorFlow at least 2.19.1")
     super().setUp()
 
   @jtu.ignore_warning(