code: def fft(x):
      complex_input = _get_complex_tensor_from_tuple(x)
      complex_output = jnp.fft.fft(complex_input)
      return jnp.real(complex_output), jnp.imag(complex_output)
  
imports: import math
  import jax
  import jax.numpy
  from keras.backend import standardize_dtype
  from keras.backend.jax.core import convert_to_tensor
  from keras.utils.module_utils import scipy
label_desc: Use image node_id 9 for calling a global function with example usage: fft(x) and returns: jnp,  jnp