jax.experimental.key_reuse.reuse_key

jax.experimental.key_reuse.reuse_key(key)

Explicitly mark a key as unconsumed.

Outside the context of key reuse checking (see jax.experimental.key_reuse) this function operates as an identity.

Example

>>> import jax
>>> key = jax.random.key(0)
>>> data = jax.random.uniform(key)
>>> same_data = jax.random.uniform(reuse_key(key))