Simple Tetris Network in Torch 2.0 + C++ inference

[ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation

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

[GRaM at ICML'24] The Price of Freedom: Exploring Tradeoffs between Expressivity and Computational Efficiency in Equivariant Tensor Products

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

[ArXiv'24] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields

Your favorite basis functions for equivariant models in one place. Code borrowed from e3x

Library for computing, inverting, and visualizing spectra of local atomic environments