jax.enable_checks

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

用於 jax_enable_checks 設定選項的上下文管理器。

開啟 JAX 內部機制的恆定性檢查。會使速度變慢。

參數:

new_val (Any)