jax.random.key_dtype

Contents

jax.random.key_dtype#

jax.random.key_dtype(impl_spec=None)[source]#

Get the dtype corresponding to a PRNG implementation.

Parameters:

impl_spec (PRNGSpecDesc | None)

Return type:

prng.KeyTy