`jax.extend.random` module#

`define_prng_impl`(*, key_shape, seed, split, ...)	param key_shape:
`seed_with_impl`(impl, seed)	param impl:
`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.

jax.extend.random module