code: def skip_assert_for_tracers(f):
      """Function decorator that returns None if JAX tracers are detected."""
      if not JAX_MODE:
          return f
      from jax import (
          core as jax_core,
      )  # pylint: disable=g-import-not-at-top
  
      def wrapped(*args, **kwargs):
          if isinstance(
              np.zeros([0]), jax_core.Tracer
          ) or any(  # omnistaging
              isinstance(arg, jax_core.Tracer)
              for arg in args + tuple(kwargs.values())
          ):
              return None
          return f(*args, **kwargs)
  
      return wrapped
  
imports: import numpy
  import six
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import dtype
  from tensorflow_probability.python.internal.backend.numpy import ops
label_desc: Use image node_id 1 for calling a global function with example usage: skip_assert_for_tracers(f) and returns: wrapped, f, f, None