Add out_sharding docstrings to linear layer call methods#5154

jackopenn wants to merge 1 commit intogoogle:maingoogle/flax:mainfrom

jackopenn:linear-out-sharding-docsjackopenn/flax:linear-out-sharding-docsCopy head branch name to clipboard