enerzyme.models.schnet.core

Description: Code is adapted from torch geometric implementation pyg-team/pytorch_geometric. All rights reserved to original authors.

Classes

SchNetCore([hidden_channels, dim_embedding, ...])