visualization

Contents

visualization#

flax.nnx.display(*args)[source]#

Display the given objects using a Penzai visualizer.

If Penzai is not installed or the code is not running in IPython, display will print the objects instead.