A short (~50 lines of code) and easy-to-follow implementation of E(n) Equivariant Graph Neural Networks @vgsatorras @emiel_hoogeboom @wellingmax for HOMO energy prediction. github.com/senya-ashukha/…. Quite a fun weekend project!
1
99
617
0
202
Download Image
It is pure PyTorch & Numpy. No external packages like PyTorch-geometric.