jax.checking_leaks

jax.checking_leaks = functools.partial(<jax._src.config.State object>, True)

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

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