Use a uniform rng rather than a normal rng to defeat CSE. (#2000)
Use a uniform rng rather than a normal rng to defeat CSE. (#2000)

The normal distribution is relatively expensive to compute.
google/jax,google/jax,tensorflow/probability,google/jax,tensorflow/probability,google/jax