flax
updated `nnx.graph` docstrings
#3958
Merged

Loading