search

choderalab / mtenn

Modular Training and Evaluation of Neural Networks
MIT License
5 stars 1 forks source link

Implement ViSNet from pytorch geometric #40

Closed fyng closed 5 months ago

fyng commented 6 months ago

Implementing ViSNet model from pytorch geometric. ViSNet is an equivariant message passing neural net for molecular property prediction.

REQUIRES pytorch nightly!