A repository of building blocks in PyTorch for E(3)/SE(3)-equivariant neural networks, built on top of e3nn:
- Equivariant Convolution:
e3tools.nn.Conv - Equivariant Multi-Layer Perceptrons (MLPs):
e3tools.nn.EquivariantMLP - Equivariant Layer Norm:
e3tools.nn.LayerNorm - Equivariant Activations:
e3tools.nn.Gateande3tools.nn.Gated - Separable Equivariant Tensor Products:
e3tools.nn.SeparableTensorProduct - Extracting Irreps:
e3tools.nn.ExtractIrreps - Self-Interactions:
e3tools.nn.LinearSelfInteraction
pip install e3tools