tfep.loss

Loss functions to train PyTorch normalizing flows for reweighting.

Classes

BoltzmannKLDivLoss([ignore_nan])

KL divergence between two Boltzmann distributions.