Equivariant Graph Neural Network Python Packages

atom-jepa

Atom-JEPA: Joint-Embedding Predictive Architecture for 3D Atomistic Systems

mldft

STRUCTURES25 - Machine-learning OF-DFT implementation using equivariant graph neural networks (JACS 2025).