enerzyme.models.schnet# Modules core Description: Code is adapted from torch geometric implementation pyg-team/pytorch_geometric. interaction