jax.enable_custom_prng

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

对 jax_enable_custom_prng 配置选项的上下文管理器（临时）。

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

参数:

new_val (任意类型)