Temporarily disable flaky nextafter tests

These are currently failing at HEAD due to llvm/llvm-project@72f10f7

We can re-enable once llvm/llvm-project@b9483d3 is integrated.

PiperOrigin-RevId: 601788984

jax/_src/internal_test_util/lax_test_util.py

@@ -160,13 +160,14 @@ def lax_ops():
       op_record("floor", 1, float_dtypes, test_util.rand_small),
       op_record("ceil", 1, float_dtypes, test_util.rand_small),
       op_record("round", 1, float_dtypes, test_util.rand_default),
-      op_record(
-          "nextafter",
-          2,
-          [f for f in float_dtypes if f != dtypes.bfloat16],
-          test_util.rand_default,
-          tol=0,
-      ),
+      # TODO(b/322390905) re-enable this test
+      # op_record(
+      #     "nextafter",
+      #     2,
+      #     [f for f in float_dtypes if f != dtypes.bfloat16],
+      #     test_util.rand_default,
+      #     tol=0,
+      # ),
       op_record("is_finite", 1, float_dtypes, test_util.rand_small),
       op_record("exp", 1, float_dtypes + complex_dtypes, test_util.rand_small),
       op_record("exp2", 1, float_dtypes + complex_dtypes, test_util.rand_small),

tests/lax_numpy_operators_test.py

@@ -124,8 +124,9 @@ def op_record(name, nargs, dtypes, shapes, rng_factory, diff_modes,
     op_record("minimum", 2, all_dtypes, all_shapes, jtu.rand_some_inf, []),
     op_record("multiply", 2, all_dtypes, all_shapes, jtu.rand_default, ["rev"]),
     op_record("negative", 1, number_dtypes, all_shapes, jtu.rand_default, ["rev"]),
-    op_record("nextafter", 2, [f for f in float_dtypes if f != jnp.bfloat16],
-              all_shapes, jtu.rand_default, ["rev"], inexact=True, tolerance=0),
+    # TODO(b/322390905) re-enable this test
+    # op_record("nextafter", 2, [f for f in float_dtypes if f != jnp.bfloat16],
+    #           all_shapes, jtu.rand_default, ["rev"], inexact=True, tolerance=0),
     op_record("not_equal", 2, all_dtypes, all_shapes, jtu.rand_some_equal, ["rev"]),
     op_record("array_equal", 2, number_dtypes, all_shapes, jtu.rand_some_equal, ["rev"]),
     op_record("array_equiv", 2, number_dtypes, all_shapes, jtu.rand_some_equal, ["rev"]),