flax

3a5db4d7 - Merge pull request #684 from jheek:fix-non-init-module-attr