visualization# flax.nnx.display(*args)[source]# Display the given objects using the Treescope pretty-printer. If treescope is not installed or the code is not running in IPython, display will print the objects instead.
