schnetpack.model

AtomisticModel	Base class for all SchNetPack models.
NeuralNetworkPotential	A generic neural network potential class that sequentially applies a list of input modules, a representation module and a list of output modules.