jax.enable_custom_prng¶

jax.enable_custom_prng = <jax._src.config._StateContextManager object>¶

Context manager for jax_enable_custom_prng config option (transient).

Enables an internal upgrade that allows one to define custom pseudo-random number generator implementations. This will be enabled by default in future versions of JAX, at which point disabling it will be considered deprecated. In a version after that the flag will be removed altogether.