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.