jax.extend.random module — JAX documentation (original) (raw)
jax.extend.random module#
| define_prng_impl(*, key_shape, seed, split, ...) | |
|---|---|
| seed_with_impl(impl, seed) | |
| threefry2x32_p | |
| threefry_2x32(keypair, count) | Apply the Threefry 2x32 hash. |
| threefry_prng_impl | Specifies PRNG key shape and operations. |
| rbg_prng_impl | Specifies PRNG key shape and operations. |
| unsafe_rbg_prng_impl | Specifies PRNG key shape and operations. |