visualization# flax.experimental.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.
