jax.enable_checks

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

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

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

参数:

new_val (Any)