Migrate GAT from DGL to Pytorch Geometric

choderalab / mtenn

Modular Training and Evaluation of Neural Networks

MIT License

5 stars 1 forks source link

Open kaminow opened 2 months ago

kaminow commented 2 months ago

Looks like PyG has a GAT model too, not sure why I didn't use this one from the beginning.

hmacdope commented 2 months ago

This would be handy to hopefully remove DGL as a very tricky dep.

kaminow commented 2 months ago

I can start devoting some bandwidth to this now that docs are done

hmacdope commented 2 months ago

Its not super urgent, but would be handy long term (including for MTENN)