code: def hide_flax_in_tracebacks():
      """Hides Flax internal stack frames in tracebacks."""
      global _flax_exclusions, _flax_filter_tracebacks
      _flax_filter_tracebacks = True
      for exclusion in _flax_exclusions:
          if exclusion not in jax_traceback_util._exclude_paths:
              jax_traceback_util._exclude_paths.append(exclusion)
  
imports: from jax._src import traceback_util
  from flax import config
label_desc: Use image node_id 2 for calling a global function with example usage: hide_flax_in_tracebacks() without return types