Log probability of samples using NUTS / HMC

Hi @tcbegley, for the first question, you can set extra_fields=('potential_energy',) in MCMC.run, then call the method MCMC.get_extra_fields. For the second question, I think you can use log_density or potential_energy utility, depending on your posterior values are in constrained space or unconstrained space (see the docs of those functions for more explanation).

1 Like