search

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

Angles2Backbone further optimization #12

Closed lupoglaz closed 5 years ago

lupoglaz commented 5 years ago

Forward pass can still be accelerated using GPU scan algorithms, summation with grad output should be accelerated with GPU reduce. We still can cut the backward time almost in half using these tricks.

lupoglaz commented 5 years ago

This layer is not a bottleneck now. The code quality improved significantly in version 0.15. Closing this issue.