jax.enable_custom_prng

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

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

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

参数:

new_val (Any)