https://github.com/google/jax
Raw File
Tip revision: 48ad9a6f3e72f19f5b86f3bb014587e6bc6c0c4c authored by Yash Katariya on 31 May 2023, 23:48:13 UTC
Start jax and jaxlib 0.4.11 release
Tip revision: 48ad9a6
jax.random.rst
``jax.random`` module
=====================

.. automodule:: jax.random

List of Available Functions
---------------------------

.. Generate the list below as follows:
   >>> from jax import random
   >>> fns = (x for x in sorted(dir(random)) if x != 'threefry_2x32')
   >>> fns = (x for x in fns if callable(getattr(random, x)))
   >>> print('\n'.join('    ' + x for x in fns))  # doctest: +SKIP

.. autosummary::
  :toctree: _autosummary

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

back to top