jax.random.rst

jax.random package

.. automodule:: jax.random

List of Available Functions

.. autosummary::
  :toctree: _autosummary

    PRNGKey
    bernoulli
    beta
    categorical
    cauchy
    choice
    dirichlet
    double_sided_maxwell
    exponential
    fold_in
    gamma
    gumbel
    laplace
    logistic
    maxwell
    multivariate_normal
    normal
    pareto
    permutation
    poisson
    rademacher
    randint
    shuffle
    split
    t
    truncated_normal
    uniform
    weibull_min