Merge branch 'main' of https://github.com/prescient-design/e3tools

Author: ameya98

README.md

@@ -1,14 +1,17 @@
 # e3tools
 
+**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).**
+
 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):
 - Equivariant Linear Layers: [`e3tools.nn.Linear`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_linear.py#L3)
-- 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)
+- 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)
 - Equivariant Multi-Layer Perceptrons (MLPs): [`e3tools.nn.EquivariantMLP`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_mlp.py#L86)
 - Equivariant Layer Norm: [`e3tools.nn.LayerNorm`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_layer_norm.py#L9)
 - Equivariant Activations: [`e3tools.nn.Gate`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_gate.py#L10), [`e3tools.nn.GateWrapper`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_gate.py#L117) and [`e3tools.nn.Gated`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_gate.py#L68)
 - Separable Equivariant Tensor Products: [`e3tools.nn.SeparableTensorProduct`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_tensor_product.py#L8)
 - Extracting Irreps: [`e3tools.nn.ExtractIrreps`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_extract_irreps.py#L5)
 - Self-Interactions: [`e3tools.nn.LinearSelfInteraction`](https://github.com/prescient-design/e3tools/blob/main/src/e3tools/nn/_interaction.py#L5)
+- 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)
 
 All modules are compatible with `torch.compile` for JIT compilation.