jax.enable_custom_prng

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

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

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

参数:

new_val (任意类型)