code: def log_compiles_config():
      """Setup and teardown of jax_log_compiles flag"""
      with jax.log_compiles(True):
          yield
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 8 for calling a global function with example usage: log_compiles_config() without return types