jax.log_compiles¶

jax.log_compiles = <jax._src.config._StateContextManager object>¶

Context manager for jax_log_compiles config option.

Log a message each time every time jit or pmap compiles an XLA computation. Logging is performed with absl.logging. When this option is set, the log level is WARNING; otherwise the level is DEBUG.