jax.enable_checks

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

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

开启 JAX 内部的不变性检查。这会降低速度。

参数:

new_val (任意类型)