flax
[nnx] update State and variables docstrings
#4346
Merged

Loading