terminator.models.layers.term.graph.s2s

TERM MPNN modules

This file contains Attention and Message Passing implementations of the TERM MPNN.

Classes

TERMEdgeEndpointAttention(num_hidden, num_in)	TERM Edge Endpoint Attention
TERMEdgeMPNNLayer(num_hidden, num_in[, ...])	TERM Edge MPNN Layer
TERMEdgeTransformerLayer(num_hidden, num_in)	TERM Edge Transformer Layer
TERMGraphTransformerEncoder(hparams)	TERM Graph Transformer Encoder
TERMNeighborAttention(num_hidden, num_in[, ...])	TERM Neighbor Attention
TERMNodeMPNNLayer(num_hidden, num_in[, ...])	TERM Node MPNN Layer
TERMNodeTransformerLayer(num_hidden, num_in)	TERM Node Transformer Layer