jax.enable_custom_prng

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

用於 jax_enable_custom_prng 配置選項的上下文管理器（暫時性）。

啟用內部升級，允許使用者定義自訂偽隨機數產生器實作。這將在未來版本的 JAX 中預設啟用，屆時所有使用此標誌的情況都將被視為已棄用（遵循 API 相容性政策）。

參數:

new_val (Any)