jax.enable_custom_prng

jax.enable_custom_prng = <jax._src.config.State object>

jax_enable_custom_prng 配置选项的上下文管理器（瞬态）。

启用一个内部升级，允许用户定义自定义伪随机数生成器实现。 这将在 JAX 的未来版本中默认启用，届时所有对该标志的使用都将被视为已弃用（遵循 API 兼容性策略）。

参数:

new_val (任意)