flax

2ee10c2b - Merge pull request #5442 from samanklesaria:fix-rnglib-doctest