lamoureux-lab / TorchProteinLibrary

PyTorch library of layers acting on protein representations
https://lamoureux-lab.github.io/TorchProteinLibrary/
MIT License
116 stars 23 forks source link

FullAtomModel on GPU #17

Open lupoglaz opened 5 years ago

lupoglaz commented 5 years ago

In principle we need Angles2Coords and other FullAtomModel layers working on GPU. The only complication is that we use recursive depth-first graph traversal which is not suited for vector processors. If this traversal can be effectively unrolled the computation time seems to benefit immensely.