code: def if_else(cond, body_fn, orelse_fn, vars):
      cond_vars = list(vars.values())
      pred = cond(**vars)
      with jax.disable_jit():
          final_vars = jax.lax.cond(
              pred, body_fn, orelse_fn, *cond_vars
          )
      return final_vars
  
label_desc: Use image node_id 1 for calling a global function with example usage: if_else(cond, body_fn, orelse_fn, vars) and returns: final_vars