• Stars
    star
    113
  • Rank 310,115 (Top 7 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created almost 3 years ago
  • Updated almost 3 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks

Simple implementation of Equivariant GNN

Open In Colab

  • A short implementation of E(n) Equivariant Graph Neural Networks for HOMO energy prediction.
  • Just 50 lines of code;
  • The implementation is based on pure PyTorch & Numpy, it has no external packages (like PyTorch-geometric).
  • Closely matches the Mean Absolute Error reported in the paper.