jax

3f59fa68 - Add replace option to random.categorical to enable sampling without replacement.