Curious to try, thanks for putting this out.
What did you find to be the most surprising challenge in this project?
^How convoluted MSA and template handling is under the hood in openfold. I’m working on a simple offline replacement next. Compared to that, the attention functions were fairly easy to convert to MLX ops.
^How convoluted MSA and template handling is under the hood in openfold. I’m working on a simple offline replacement next. Compared to that, the attention functions were fairly easy to convert to MLX ops.