一种简化的 PyTorch 实现版本，用于 AlphaFold2 模型与训练。
A minimal PyTorch re-implementation of AlphaFold2's model & training
