Add split and key methods to RngStream and Rngs#5047

samanklesaria wants to merge 4 commits intogoogle:maingoogle/flax:mainfrom

samanklesaria:issues/5046samanklesaria/flax:issues/5046Copy head branch name to clipboard