You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+4-1Lines changed: 4 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,14 +1,17 @@
1
1
# e3tools
2
2
3
+
**New: We have added support for [OpenEquivariance](https://github.com/PASSIONLab/OpenEquivariance) kernels! Check out [`e3tools.nn.FusedConv`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_conv.py#L26).**
4
+
3
5
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
6
- Equivariant Linear Layers: [`e3tools.nn.Linear`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_linear.py#L3)
5
-
- 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)
7
+
- Equivariant Convolution: [`e3tools.nn.Conv`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_conv.py#L166) and [`e3tools.nn.SeparableConv`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_conv.py#L284)
- Tools for working with graphs: [`e3tools.scatter`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/_scatter.py#L17), [`e3tools.scatter_softmax`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/_scatter.py#L49) and [`e3tools.radius_graph`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/_radius.py)
12
15
13
16
All modules are compatible with `torch.compile` for JIT compilation.
0 commit comments