code: def array_split(ary, indices_or_sections, axis=0):
      return ivy.split(
          ary,
          num_or_size_splits=indices_or_sections,
          axis=axis,
          with_remainder=True,
      )
  
imports: import ivy
  from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back, handle_jax_dtype
  from ivy.functional.frontends.jax.numpy import promote_types_of_jax_inputs
label_desc: Use image node_id 2 for calling a global function with example usage: array_split(ary, indices_or_sections, axis) and returns: ivy