xnn.gnn.featurizers#
Modules
Cartesian angular edge basis for the CACE potential (Cheng 2024). |
|
Polynomial cutoff envelope used by the equivariant GNN edge embedding (NequIP/MACE). |
|
Radial pieces of the equivariant GNN edge embedding. |
|
Equivariant edge featurization for E(3) GNNs (NequIP / MACE / Allegro). |