jax.experimental.random 模块

实验性随机数 API。

stateful_rng([seed, impl])	具有隐式更新状态的实验性有状态 RNG。
StatefulPRNG(_base_key, _counter)	有状态的 JAX 随机生成器。