flax
updated `nnx.module` docstrings
#3966
Merged

Loading