jax

e675f804 - Add support for 8- and 16-bit output in _random_bits (#3090)