Skip to main content
Ctrl+K

jax.extend.random.seed_with_impl#

jax.extend.random.seed_with_impl(impl, seed)[source]#
Parameters:
  • impl (PRNGImpl) –

  • seed (Union[int, Array, ndarray, bool_, number, bool, float, complex]) –

Return type:

PRNGKeyArrayImpl

previous

jax.extend.random.define_prng_impl

next

jax.extend.random.threefry2x32_p