flax
add missing docs for module functions
#3619
Merged

Loading