可视化

flax.nnx.display(*args)

使用 Treescope 美观打印机显示给定对象。

如果未安装 treescope 或代码未在 IPython 中运行，则 display 将改为打印对象。