flax

836946a6 - Merge pull request #938 from levskaya:lazy_setup