Bump test tolerance on FFT test that started failing in CI after an X…

…LA change.

PiperOrigin-RevId: 679715691

tests/fft_test.py

@@ -161,7 +161,7 @@ def testFftn(self, inverse, real, shape, dtype, axes, s, norm):
     # Numpy promotes to complex128 aggressively.
     self._CheckAgainstNumpy(np_fn, jnp_fn, args_maker, check_dtypes=False,
                             tol=1e-4)
-    self._CompileAndCheck(jnp_fn, args_maker)
+    self._CompileAndCheck(jnp_fn, args_maker, atol={np.complex64: 2e-6})
     # Test gradient for differentiable types.
     if (config.enable_x64.value and
         dtype in (float_dtypes if real and not inverse else inexact_dtypes)):