code: def unstack(
      x: JaxArray,
      /,
      *,
      copy: Optional[bool] = None,
      axis: int = 0,
      keepdims: bool = False,
  ) -> List[JaxArray]:
      if x.shape == ():
          return [x]
      dim_size = x.shape[axis]
      # ToDo: make this faster somehow, jnp.split is VERY slow for large dim_size
      x_split = jnp.split(x, dim_size, axis)
      if keepdims:
          return x_split
      return [jnp.squeeze(item, axis) for item in x_split]
  
imports: import math
  from numbers import Number
  from typing import Union, Tuple, Optional, List, Sequence, Iterable
  import jax.numpy
  import numpy
  import ivy
  from ivy.func_wrapper import with_unsupported_dtypes
  from ivy.functional.backends.jax import JaxArray
  from .None import backend_version
label_desc: Use image node_id 15 for calling a global function with example usage: unstack() and returns: unknown, list, x_split