[deepmd.pt]
mace = deepmd_gnn.mace:MaceModel
nequip = deepmd_gnn.nequip:NequipModel

