flax

24224bff - Merge pull request #2816 from jheek:lazy-init