Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2 - View it on GitHub