jax

aea6b320 - Merge pull request #21216 from jakevdp:pmap-prng