flax.traceback_util package#

Flax specific traceback_util functions.

Traceback filtering utils#

flax.traceback_util.hide_flax_in_tracebacks()[source]#

Hides Flax internal stack frames in tracebacks.

flax.traceback_util.show_flax_in_tracebacks()[source]#

Shows Flax internal stack frames in tracebacks.