jax.enable_checks

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

用于 jax_enable_checks 配置选项的上下文管理器。

开启 JAX 内部的不变式检查。会使程序变慢。

参数:

new_val (任意类型)