flax

42980115 - Merge pull request #1557 from levskaya:rngs_check_fix