Contents:
Classes
EdgeLayer(node_dims, edge_dims[, drop_rate, ...])
EdgeLayer
GVP Edge MPNN
GVPPairEnergies(hparams)
GVPPairEnergies
GNN Potts Model Encoder using GVP