gussmith23 / glenside

A pure, low-level tensor program representation enabling tensor program optimization via program rewriting. See the web demo at https://gussmith23.github.io/glenside-web-demo/
68 stars 10 forks source link

Import Rosettafold model #174

Open gussmith23 opened 2 years ago

gussmith23 commented 2 years ago

The Rosettafold model (which is implemented in PyTorch, I think) could be a good test example for reducing data transpositions using Glenside.

https://github.com/RosettaCommons/RoseTTAFold/tree/main/network