ukiddlmZmZddlmZddlmZddlmZee dfZ dddd e d eegefd eee gefd eee e gefd eee gefde de defdZ y))CallableHashable)prng)random)Array.z ?nametag key_shapeseedsplit random_bitsfold_inr r returnc `tjtj|||||||S)Nr )rPRNGSpecrPRNGImpl)r r rrrr r s Q/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/extend/random.pydefine_prng_implrs0 uk7 S N) collections.abcrrjax._srcrrjax._src.typingrtupleintShapestrrrrr s/! c3h"- #  % #UGUN3 %eU^U%:; #+E3+>+E"F  's|U':;    )1 r