senya-ashukha / simple-equivariant-gnn Star 124 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 17 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 Jun 12, 2024 Jupyter Notebook