Correct implementation of log_accept_ratio in NUTS, as it was accumulating energy_diff from subtree that have already been rejected. Note that the bug does not impact the correctness of samples, as the energy_diff_sum is used in tuning only.
Correct implementation of log_accept_ratio in NUTS, as it was accumulating energy_diff from subtree that have already been rejected.
Note that the bug does not impact the correctness of samples, as the energy_diff_sum is used in tuning only.

PiperOrigin-RevId: 275514226

tensorflow/probability,tensorflow/probability