jax.extend.random.seed_with_impl

Contents

jax.extend.random.seed_with_impl#

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

  • seed (int | typing.ArrayLike) –

Return type:

PRNGKeyArray