jax.experimental.random module

Experimental random APIs.

stateful_rng([seed, impl])	Experimental stateful RNG with implicitly-updated state.
StatefulPRNG(_base_key, _counter)	Stateful JAX random generator.