torch-pmeΒΆ
torch-pme provides a modular implementation of classical algorithms for the evaluation of long-range interactions (e.g. Ewald, particle-mesh Ewald - PME, and P3M). It can be used to build traditional electrostatic models, as well as long-range descriptors for machine-learning applications.
Atomistic Water Model for Molecular Dynamics

Learning Capabilities with torchpme
