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). If b <= a, then the result is undefined, and different implementations may return different results.
You should use jax.random for most purposes; this function exists only for niche use cases with special performance requirements.
This API may be removed at any time.