egnn

There are 1 repository under egnn topic.

senya-ashukha / simple-equivariant-gnn
A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks
graph-neural-networks gnn equivariance pytorch egnn qm9 homo molecules neural-networks deep-learning deep-neural-networks
Language:Python 121
smsharma / eqnn-jax
Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
egnn n-body-simulation symmetry equivariant-neural-networks nequip segnn
Language:Jupyter Notebook 13

senya-ashukha / simple-equivariant-gnn