Quickstart
Guided Tour
How do I ...?
Design Notes
setup
nn.compact
Additional material
API reference
Flax specific traceback_util functions.
Hides Flax internal stack frames in tracebacks.
Shows Flax internal stack frames in tracebacks.