jax.enable_checks

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

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

打开 JAX 内部的不可变性检查。这会使速度变慢。

参数:

new_val (Any)