torch-geometric
Graph Neural Networks (PyG). Node/graph classification, link prediction, GCN, GAT, GraphSAGE, heterogeneous graphs, molecular property prediction, for geometric deep learning.
graph-neural-networkspytorchgeometric-deep-learningnode-classification
ovachiever
81
equivariant-architecture-designer
Use when you have validated symmetry groups and need to design neural network architecture that respects those symmetries. Invoke when user mentions equivariant layers, G-CNN, e3nn, steerable networks, building symmetry into model, or needs architecture recommendations for specific symmetry groups. Provides architecture patterns and implementation guidance.
machine-learningarchitectural-patternsgeometric-deep-learningequivariant-layers
lyndonkl
82
torch-geometric
Graph Neural Networks (PyG). Node/graph classification, link prediction, GCN, GAT, GraphSAGE, heterogeneous graphs, molecular property prediction, for geometric deep learning.
pythonmachine-learningpytorchgraph-neural-networks
K-Dense-AI
3,233360