flax

cd1d5313 - Merge pull request #469 from jheek:linen-docs-udpate