Use a faster, numerically more faithful, approach to logaddexp
Use a faster, numerically more faithful, approach to logaddexp

We can use log1p and fewer instances of exp to compute the same result.

google/jax,google/jax,tensorflow/probability,google/jax,google/jax,tensorflow/probability