Instantiate RNG in testcase instead of test harness in a few more places. (#2706)
Instantiate RNG in testcase instead of test harness in a few more places. (#2706)


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