geometric-deep-learning

Graph Neural Network Library for PyTorch

Drench yourself in Deep Learning, Reinforcement Learning, Machine Learning, Computer Vision, and NLP by learning from these exciting lectures!!

StellarGraph - Machine Learning on Graphs

Source code for GNN-LSPE (Graph Neural Networks with Learnable Structural and Positional Representations), ICLR 2022

A Transformer made of Rotation-equivariant Attention using Vector Neurons

Implementation of ICLR23 paper "Molecule Generation for Target Protein Binding with Structural Motifs"

A geometric deep learning framework (Geometric Transformers) for predicting protein interface contacts. (ICLR 2022)

Platonic Transformers: A Solid Choice For Equivariance (ICML 2026)