flax

79851d4f - Merge pull request #3767 from google:improve-make_rng-docstring