torch-pme

torch-pme is a library that enables efficient and auto-differentiable computation of long-range interactions in molecular systems using PyTorch. It supports various methods for calculating electrostatic interactions, making it suitable for applications in molecular dynamics and simulations.