code: def beta(shape, alpha, beta, dtype=None, seed=None):
      dtype = dtype or floatx()
      seed = jax_draw_seed(seed)
      # jax doesn't accept python lists as arguments
      alpha = jax.numpy.array(alpha)
      beta = jax.numpy.array(beta)
      sample = jax.random.beta(
          key=seed, a=alpha, b=beta, shape=shape, dtype=dtype
      )
      return sample
  
file_path: keras/keras/backend/jax/random.py
imports: import jax
  from keras.backend.config import floatx
  from keras.random.seed_generator import SeedGenerator
  from keras.random.seed_generator import draw_seed
  from keras.random.seed_generator import make_default_seed
label_desc: Use image node_id 12 for calling a global function with example usage: beta(shape, alpha, beta, dtype, seed) and returns: sample