code: def unsorted_segment_min(
      data: JaxArray,
      segment_ids: JaxArray,
      num_segments: int,
  ) -> JaxArray:
      # added this check to keep the same behaviour as tensorflow
      ivy.utils.assertions.check_unsorted_segment_valid_params(
          data, segment_ids, num_segments
      )
      return jax.ops.segment_min(data, segment_ids, num_segments)
  
file_path: ivy/ivy/functional/backends/jax/experimental/creation.py
imports: from typing import Optional, Tuple
  import math
  import jax
  import jax.numpy
  import jaxlib.xla_extension
  from ivy.functional.backends.jax import JaxArray
  import ivy
label_desc: Use image node_id 5 for calling a global function with example usage: unsorted_segment_min(data, segment_ids, num_segments) and returns: jax