File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11# e3tools
22
33A repository of building blocks in PyTorch 2.0 for E(3)/SE(3)-equivariant neural networks, built on top of [ e3nn] ( https://github.com/e3nn/e3nn ) :
4+ - Equivariant Linear Layers: [ ` e3tools.nn.Linear ` ] ( https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_linear.py#L3 )
45- Equivariant Convolution: [ ` e3tools.nn.Conv ` ] ( https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_conv.py#L16 ) and [ ` e3tools.nn.SeparableConv ` ] ( https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_conv.py#L124 )
56- Equivariant Multi-Layer Perceptrons (MLPs): [ ` e3tools.nn.EquivariantMLP ` ] ( https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_mlp.py#L86 )
67- Equivariant Layer Norm: [ ` e3tools.nn.LayerNorm ` ] ( https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_layer_norm.py#L9 )
You can’t perform that action at this time.
0 commit comments