jax.random

E438359

jax.random is JAX’s module for generating and manipulating pseudo-random numbers in a functional, reproducible way using PRNG keys.

All labels observed (1)

Label Occurrences
jax.random canonical 1

How this entity was disambiguated

Statements (60)

Predicate Object
instanceOf JAX submodule
Python module
compatibleWith jax.grad
jax.jit
jax.pmap
jax.vmap
designedFor pure functional style
distributedBy JAX project NERFINISHED
documentationURL https://jax.readthedocs.io/en/latest/jax.random.html
ensures explicit handling of random state
stateless random number generation
hostedOn GitHub (google/jax) NERFINISHED
implementedIn Python NERFINISHED
implements pseudo-random number generation
partOf jax NERFINISHED
providesFunction PRNGKey
bernoulli
beta
binomial
categorical
cauchy
chisquare
choice
default_prng_impl
dirichlet
exponential
f
fold_in
gamma
geometric
gumbel
key
laplace
logistic
lognormal
logseries
multivariate_normal
normal
orthogonal
pareto
permutation
poisson
rademacher
randint
seed
shuffle
split
t
threefry2x32
threefry_2x32
triangular
truncated_exponential
truncated_normal
uniform
vonmises
wald
weibull_min
requires PRNGKey
supports reproducible random number generation
uses functional PRNG model

How these facts were elicited

Referenced by (1)

Full triples — surface form annotated when it differs from this entity's canonical label.

JAX hasComponent jax.random