jax.log_compiles

Contents

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 logging. When this option is set, the log level is WARNING; otherwise the level is DEBUG.