Alex wrote a Triton kernel that allows the model to scale to thousands of residues! Runs faster and is more memory efficient. Huge contribution, check it out on the repo!