jax

6c56d651 - jax.random.bernoulli: add mode='high' for improved sampling for small p