jax.extend.random module

`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.
`random_seed`(seeds, impl)
`rbg_prng_impl`	Specifies PRNG key shape and operations.
`unsafe_rbg_prng_impl`	Specifies PRNG key shape and operations.