TorchMD-Net implements trainable neural network potentials as PyTorch modules, with TensorNet architectures, custom molecular datasets, pretrained checkpoint loading and molecular dynamics integrations.
Open sourcePython
Molecular Property Prediction · Computational Chemistry