equivariant-neural-networks topic

Tools for exploiting Morphological Symmetries in robotics

Danfoa

Library to make any existing neural network architecture equivariant

arnab39

[ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs

atomicarchitects

[ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations

atomicarchitects

Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.

smsharma