e3tools documentation

e3tools is a repository of building blocks in PyTorch for E(3)/SE(3)-equivariant neural networks.

Contents:

• e3tools package
  • Subpackages
    • e3tools.nn package
  • Module contents
    • pack_irreps()
    • radius()
    • radius_graph()
    • scatter()
    • unpack_irreps()