Sparsifying OpenFold (a structural biology deep learning model)

[engelberger]

Hi! I am interested in sparsifying and quantizing a structural biology deep learning model as part of my PhD research, is it possible? if can you provide me any particular tutorial/ overview on where should I start? Having a lightweight model would be of great help in my project.

The model in question:
https://github.com/aqlaboratory/openfold

Thank you in advance!

[mgoin]

Hi @engelberger I think it's possible! I could help you do some performance estimates if you can find a way to export an ONNX model. PyTorch models are generally pretty successful in converting to ONNX