Add `weighted_sample_logps`.

[joshrule]

It would be nice to create a version of weighted_sample that takes in log probabilities. This might require the uniform to exponential trick described here.