Models

Gate-points 2102

Convolution(irreps_in, irreps_node_attr, ...)	Equivariant edge convolution with radial TensorProduct weights.
Network(irreps_in, irreps_hidden, ...[, ...])	Gated E(3)-equivariant network operating on one or more point graphs.

Modular v2106 models

Convolution(irreps_node_input, ...[, ...])	v2106 convolution with a learned scalar residual mixing factor.
MessagePassing(irreps_node_sequence, ...[, ...])	Sequence of v2106 convolutions and parity-aware gates.
SimpleNetwork(irreps_in, irreps_out, ...[, ...])	v2106 point network that derives all geometric edge attributes from positions.
NetworkForAGraphWithAttributes(...[, mul, ...])	v2106 point network with caller-provided node and edge attributes.