lloca.equivectors.lgatr

Edge convolution with L-GATr.

Classes

`LGATrSlimVectors`(n_vectors, num_scalars, ...)	Wrapper around the vector-stream `lgatr.nets.LGATrSlim` backbone.
`LGATrVectors`(n_vectors, num_scalars, ...[, ...])	Wrapper around the multivector `lgatr.nets.LGATr` backbone.

class lloca.equivectors.lgatr.LGATrSlimVectors(n_vectors, num_scalars, hidden_v_channels, hidden_s_channels, net, operation='add', nonlinearity='softmax', aggr='sum', layer_norm=False, lgatr_norm=True, use_amp=False, attention_backend='xformers')[source]

Wrapper around the vector-stream lgatr.nets.LGATrSlim backbone.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

class lloca.equivectors.lgatr.LGATrVectors(n_vectors, num_scalars, hidden_mv_channels, hidden_s_channels, net, operation='add', nonlinearity='softmax', aggr='sum', layer_norm=False, lgatr_norm=True, use_amp=False, attention_backend='xformers')[source]

Wrapper around the multivector lgatr.nets.LGATr backbone.

Initialize internal Module state, shared by both nn.Module and ScriptModule.