code: def empty_stack(
      capacity: int, prng_key: ndarray = TAIL_DEFAULT_PRNGKEY
  ) -> CrayTail:
      """Returns a tuple with the size of the stack and a stack array."""
      return CrayTail(
          jnp.array([capacity]),
          randint(prng_key, (capacity,), 0, TAIL_MAX, TAIL_DTYPE),
      )
  
imports: from typing import NamedTuple, Sequence, Tuple
  import jax.lax
  import jax.numpy
  from jax.numpy import ndarray
  from jax.random import PRNGKey, randint
label_desc: Use image node_id 2 for calling a global function with example usage: empty_stack(capacity, prng_key) and returns: CrayTail