code: def jit(*args, **kwargs):
      """Just-In-Time compiles the given function for use on accelerators."""
      global _disable_jit
      if _disable_jit:
          return args[0]  # jit(f, **unused_now_jit_kwargs) = f
      return backend()["jit"](*args, **kwargs)
  
imports: import contextlib
  import enum
  import gin
  from trax.fastmath.jax import JAX_BACKEND
  from trax.fastmath.numpy import NUMPY_BACKEND
  from trax.fastmath.tf import TF_BACKEND
label_desc: Use image node_id 26 for calling a global function with example usage: jit() and returns: backend, args