flax

5d776fb2 - Merge pull request #4899 from google:jax-random-methods-docs