flax

0e5a0b4d - Merge pull request #4281 from google:improve-randomness-guide