jaxvacua.util.random_integer_jit

random_integer_jit(rns_key, lower_bound, upper_bound, shape=(1,))

JIT-compiled version of random_integer().

Parameters:

• rns_key (Array) – JAX PRNG-key array.

• lower_bound (int) – Lower edge of the sampling interval.

• upper_bound (int) – Upper edge of the sampling interval (inclusive).

• shape (Tuple[int, ...]) – Output shape. Default (1,).

Returns:

Array – Integer samples of shape shape.

Return type:

Array