flax

79ff138d - Merge pull request #779 from avital:add-make-rng