jax.random.PRNGKey#

jax.random.PRNGKey(seed, *, impl=None)[source]#

Create a pseudo-random number generator (PRNG) key given an integer seed.

The resulting key carries the default PRNG implementation, as determined by the optional impl argument or, otherwise, by the jax_default_prng_impl config flag.

Parameters:
Return type:

Array

Returns:

A PRNG key, consumable by random functions as well as split and fold_in.