flax

38cd9083 - Merge pull request #4876 from google:rngs-helpers