jax.enable_custom_prng

内容

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

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

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