code: def unique_counts(x, /):
      """Returns the unique elements of an input array x and the corresponding counts for each unique element in x."""
      return jax.numpy.unique_counts(x)
  
imports: import jax
label_desc: Use image node_id 2 for calling a global function with example usage: unique_counts() and returns: jax