tfep.io

Modules

dataset

log

Utility classes to store potential energies and CVs necessary for TFEP.

sampler

A PyTorch stateful batch sampler that can be used to correctly resume training mid-epoch.