Improve robustness of ed RVs in mixed-precision settings.

Currently, ed RandomVariable sampling for reparameterized distributions can encounter errors in mixed-precision settings (such as bfloat16) due to a mismatch between the the dtype of output of the `sample`computation in the base distribution and the dtype of the RV `Variable` parameters that are combined with the sample via multiplication/addition.