jax.检查_追踪器_泄漏

jax.检查_追踪器_泄漏#

jax.check_tracer_leaks = <jax._src.config.State object>#

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

在跟踪完成后立即开启泄漏跟踪检查。启用泄漏检查可能会影响性能：某些缓存被禁用，并且可能会增加其他开销。此外，请注意某些Python调试器可能会导致误报，因此建议在启用泄漏检查时禁用任何调试器。