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.
