Skip to content

Commit c45f72f

Browse files
authored
Add Linear to README
1 parent edc859b commit c45f72f

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# e3tools
22

33
A 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)

0 commit comments

Comments
 (0)