code: def randint(
      low: Union[int, JaxArray],
      high: Union[int, JaxArray],
      /,
      *,
      shape: Optional[Union[ivy.NativeShape, Sequence[int]]] = None,
      device: jaxlib.xla_extension.Device = None,
      dtype: Optional[Union[jnp.dtype, ivy.Dtype]] = None,
      seed: Optional[int] = None,
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      if not dtype:
          dtype = ivy.default_int_dtype()
      dtype = ivy.as_native_dtype(dtype)
      _randint_check_dtype_and_bound(low, high, dtype)
      shape = _check_bounds_and_get_shape(low, high, shape).shape
  
      if seed:
          rng_input = jax.random.PRNGKey(seed)
      else:
          RNG_, rng_input = jax.random.split(_getRNG())
          _setRNG(RNG_)
  
      return jax.random.randint(rng_input, shape, low, high, dtype)
  
file_path: ivy/ivy/functional/backends/jax/random.py
imports: import jax
  import jax.numpy
  import jaxlib.xla_extension
  from typing import Optional, Union, Sequence
  import ivy
  from ivy.functional.ivy.random import _check_bounds_and_get_shape, _randint_check_dtype_and_bound, _check_valid_scale
  from ivy.functional.backends.jax import JaxArray
  from ivy.func_wrapper import with_unsupported_dtypes
  from .None import backend_version
label_desc: Use image node_id 6 for calling a global function with example usage: randint() and returns: jax