schnetpack.representation

Message-passing neural networks

SchNet

SchNet architecture for learning representations of atomistic systems

PaiNN

PaiNN - polarizable interaction neural network