code: def vmap(
      fun,
      in_axes=0,
      out_axes=0,
      axis_name=None,
      axis_size=None,
      spmd_axis_name=None,
  ):
      fun = outputs_to_native_arrays(fun)
      return to_ivy_arrays_and_back(
          outputs_to_ivy_arrays(
              ivy.vmap(fun, in_axes=in_axes, out_axes=out_axes)
          )
      )
  
imports: import ivy
  from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back, outputs_to_native_arrays
  from ivy.func_wrapper import outputs_to_ivy_arrays
label_desc: Use image node_id 3 for calling a global function with example usage: vmap(fun, in_axes, out_axes, axis_name, axis_size, spmd_axis_name) and returns: to_ivy_arrays_and_back