flax

docs(nnx): add missing functional args to Conv and LinearGeneral

#5174

Merged