Use finfo(dtype).tiny as uniform minval
Use finfo(dtype).tiny as uniform minval

Otherwise, using the default clopen uniform for truncated_normal
introduces a slight shift of the empirical mean.

google/jax,google/jax,tensorflow/probability,google/jax,tensorflow/probability,google/jax