flax

b60f7f45 - Merge pull request #1723 from jheek:lazy-rng