jax_fresh_prng_key

probly.utils.jax.jax_fresh_prng_key() → Array

Draw a fresh PRNG key seeded from OS entropy.

Mirrors the numpy backend, where a sampler without an explicit generator falls back to a fresh numpy.random.default_rng(), so repeated calls produce different draws.

Returns:

A new PRNG key.