jax

b336daf7 - Merge pull request #28022 from jakevdp:bernoulli-fix