jax.lax.rng_uniform#

jax.lax.rng_uniform(a,b,shape)[source]#

Stateful PRNG generator. Experimental and its use is discouraged.

Returns uniformly distributed random numbers in the range [a, b). Ifb <= a, then the result is undefined, and different implementations mayreturn different results.

You should use jax.random for most purposes; this function exists only forniche use cases with special performance requirements.

This API may be removed at any time.