English

e3nn: Euclidean Neural Networks

Machine Learning 2022-07-21 v1 Artificial Intelligence Neural and Evolutionary Computing

Abstract

We present e3nn, a generalized framework for creating E(3) equivariant trainable functions, also known as Euclidean neural networks. e3nn naturally operates on geometry and geometric tensors that describe systems in 3D and transform predictably under a change of coordinate system. The core of e3nn are equivariant operations such as the TensorProduct class or the spherical harmonics functions that can be composed to create more complex modules such as convolutions and attention mechanisms. These core operations of e3nn can be used to efficiently articulate Tensor Field Networks, 3D Steerable CNNs, Clebsch-Gordan Networks, SE(3) Transformers and other E(3) equivariant networks.

Keywords

Cite

@article{arxiv.2207.09453,
  title  = {e3nn: Euclidean Neural Networks},
  author = {Mario Geiger and Tess Smidt},
  journal= {arXiv preprint arXiv:2207.09453},
  year   = {2022}
}

Comments

draft

R2 v1 2026-06-25T01:03:35.385Z