flax

d044de84 - Merge pull request #5363 from samanklesaria:rng_guide_update