jax.random: use correct x32/x64 default dtypes. (#3841)
jax.random: use correct x32/x64 default dtypes. (#3841)

This is a no-op in the current package, but will make things cleaner during the x64 deprecation.
tensorflow/probability,google/jax,google/jax,google/jax,tensorflow/probability,google/jax