flax

61be80f6 - Document `make_rng` and correct docstring