jax.random.PRNGKey — JAX documentation

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. seed (int | ArrayLike) – a 64- or 32-bit integer used as the value of the key. impl (PRNGSpecDesc | None | None) – optional string specifying the PRNG implementation (e.g. 'threefry2x32') A PRNG key, consumable by random functions as well as split and fold_in. KeyArray Previous jax.random module Next jax.random.key By The JAX authors © Copyright 2024, The JAX Authors. NumPy and SciPy documentation are copyright the respective authors..

jax.random.PRNGKey # jax.random. PRNGKey ( seed , * , impl = None ) [source] # Create a legacy PRNG key given an integer seed. This function produces old-style legacy PRNG keys, which are arrays of dtype uint32 . For more, see the note in the PRNG keys section. When possible, jax.random.key() is recommended for use instead. The resulting key does not carry a PRNG implementation. The returned key matches the implementation given by the optional impl argument or, otherwise, determined by the jax_default_prng_impl config flag. Callers must ensure that same implementation is set as the default whe

related reading

• jax.random module — JAX documentationjax.readthedocs.io
• How to think in JAX — JAX documentationdocs.jax.dev
• Key concepts — JAX documentationjax.readthedocs.io
• JAX core from scratch — JAX documentationdocs.jax.dev
• Quickstart: How to think in JAX — JAX documentationjax.readthedocs.io
• jax.Array — JAX documentationjax.readthedocs.io
• 🔪 JAX - The Sharp Bits 🔪 — JAX documentationjax.readthedocs.io
• GitHub - jax-ml/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and moregithub.com
• Why You Should (or Shouldn't) be Using Google's JAX in 2023assemblyai.com
• GPT in 60 Lines of NumPy | Jay Modyjaykmody.com
• Understanding JAX: JIT, XLA, and Pure Functions Explainedyoutube.com
• /dev/random - Wikipediaen.wikipedia.org