^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.