flax
[nnx] improve Module docs
#4499
Merged

Loading