jax.enable_检查

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

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

为 JAX 内部开启不变性检查。会使速度变慢。

参数:

new_val (Any)