jax.random.rst

jax.random module

.. automodule:: jax.random

API Reference

Key Creation & Manipulation

.. autosummary::
  :toctree: _autosummary

  key
  key_data
  wrap_key_data
  fold_in
  split
  clone
  PRNGKey

Random Samplers

.. autosummary::
  :toctree: _autosummary

    ball
    bernoulli
    beta
    binomial
    bits
    categorical
    cauchy
    chisquare
    choice
    dirichlet
    double_sided_maxwell
    exponential
    f
    gamma
    generalized_normal
    geometric
    gumbel
    laplace
    loggamma
    logistic
    lognormal
    maxwell
    multivariate_normal
    normal
    orthogonal
    pareto
    permutation
    poisson
    rademacher
    randint
    rayleigh
    t
    triangular
    truncated_normal
    uniform
    wald
    weibull_min