code: def dirichlet(
      alpha: Union[JaxArray, float, Sequence[float]],
      /,
      *,
      size: Optional[Union[ivy.NativeShape, Sequence[int]]] = None,
      dtype: Optional[jnp.dtype] = None,
      seed: Optional[int] = None,
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      if seed is not None:
          rng_input = jax.random.PRNGKey(seed)
      else:
          RNG_, rng_input = jax.random.split(_getRNG())
          _setRNG(RNG_)
      return jax.random.dirichlet(
          rng_input, alpha, shape=size, dtype=dtype
      )
  
file_path: ivy/ivy/functional/backends/jax/experimental/random.py
imports: from typing import Optional, Union, Sequence
  import jax.numpy
  import jax
  import jaxlib.xla_extension
  import ivy
  from ivy.functional.backends.jax import JaxArray
  from ivy.functional.backends.jax.random import RNG, _setRNG, _getRNG
  from ivy.functional.ivy.random import _check_bounds_and_get_shape, _check_shapes_broadcastable
  from ivy.func_wrapper import with_unsupported_dtypes
  from ..None import backend_version
label_desc: Use image node_id 1 for calling a global function with example usage: dirichlet() and returns: jax