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)

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.