ORNL / HydraGNN

Distributed PyTorch implementation of multi-headed graph convolutional neural networks
BSD 3-Clause "New" or "Revised" License
68 stars 29 forks source link

PNA Eq #280

Closed RylieWeaver closed 1 month ago

RylieWeaver commented 2 months ago

A hydrid PNA / PAINN model.

This combines our best invariant model with some comparatively "light" equivariance (meaning it has equivariance without huge architectural or computational cost). It should serve as a bridge from our models to the CG-tensor product and Spherical Harmonics models.