GitHub repository: rschireman/LightningANI
Primary language: Python
Description: Template for creating neural network potentials (NNP) with torchANI and PyTorch Lightning. Trains on both forces and energies with the use of autograd