xnn.gnn.models#
Modules
Allegro (Musaelian et al. 2023): strictly local equivariant potential. |
|
Shared bases for the GNN potentials (NequIP / MACE / Allegro / CACE). |
|
Shared E(3)-equivariant building blocks (e3nn) for the GNN potentials. |
|
CACE (Cheng 2024): Cartesian Atomic Cluster Expansion potential. |
|
MACE (Batatia et al. 2022): higher body-order equivariant message passing. |
|
Load pretrained MACE foundation models into the xnn |
|
NequIP (Batzner et al. 2022): E(3)-equivariant message-passing potential. |