torchmd-net

TorchMD-NET provides state-of-the-art neural network potentials for molecular simulations, allowing users to train and implement various architectures for efficient molecular dynamics. It integrates with existing molecular dynamics software and supports custom datasets for training.