choderalab / modelforge

Infrastructure to implement and train NNPs
https://modelforge.readthedocs.io/en/latest/
MIT License
9 stars 4 forks source link

Neural Network infrastructure work #13

Closed wiederm closed 10 months ago

wiederm commented 10 months ago

Implement a set of models that can learn energies/forces from a training dataset. We want to abstract common operations and provide them in a separate module, the NNP implementation should import these operations and apply them.