jax.log_compiles

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

Context manager for jax_log_compiles config option.

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

Parameters:

new_val (Any)