Each distribution under optax.perturbations has a method called log_prob that returns the logarithm of the unnormalized probability density at a given point.

I propose renaming this method to unnormalized_log_prob, like in TensorFlow Probability, so users don't confuse it with the actual (normalized) log probability.

If this is acceptable, I can submit a PR for it.