test `jax.extend` custom PRNG construction round trip#17988

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainfrom froystig:rng-spec-jex

Oct 6, 2023

+31

Commits on Oct 6, 2023

test custom PRNG impl construction round trip
froystig
committed