senya-ashukha / simple-equivariant-gnn Star 120 Code Issues Pull requests A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks deep-neural-networks deep-learning pytorch neural-networks homo molecules graph-neural-networks gnn equivariance qm9 egnn Updated Jan 14, 2022 Python
smsharma / eqnn-jax Star 10 Code Issues Pull requests Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP. symmetry n-body-simulation egnn equivariant-neural-networks segnn nequip Updated May 31, 2024 Jupyter Notebook