edward2 icon indicating copy to clipboard operation
edward2 copied to clipboard

Published 20 hours ago • verified publisher icon google

Improve robustness of ed RVs in mixed-precision settings.

Open dusenberrymw opened this issue 5 years ago • 0 comments

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 samplecomputation in the base distribution and the dtype of the RV Variable parameters that are combined with the sample via multiplication/addition.

dusenberrymw avatar Jan 27 '20 16:01 dusenberrymw