jax.no_tracing

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

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

禁止 JIT 编译的跟踪。

参数:

new_val (任意类型)