This repository aims to replicate AlphaFold2, DeepMind’s groundbreaking attention-based neural network that solved the 50-year grand challenge of protein structure prediction. It implements the trunk architecture using axial attention with PyTorch, supporting inputs of amino acid sequences and multiple sequence alignments. A contributor has reported training a modified trunk with competitive results against trRosetta.