Fix minor bug in bernoulli.pmf (#2221)
Fix minor bug in bernoulli.pmf (#2221)

np.exp(pmf(...)) -> np.exp(logpmf(...))
google/jax,google/jax,google/jax,tensorflow/probability,tensorflow/probability,google/jax