einops>=0.3
numpy
torch>=1.6
mdtraj>=1.8
tqdm
alphafold2-pytorch
