jax.extend.random 模块

`jax.extend.random` 模块#

`define_prng_impl`(*, key_shape, seed, split, ...)
`seed_with_impl`(impl, seed)
`threefry2x32_p`
`threefry_2x32`(keypair, count)	应用 Threefry 2x32 哈希。
`threefry_prng_impl`	指定 PRNG 密钥形状和操作。
`rbg_prng_impl`	指定 PRNG 密钥形状和操作。
`unsafe_rbg_prng_impl`	指定 PRNG 密钥形状和操作。