einops>=0.3
numpy
torch>=1.6
mdtraj>=1.8
se3-transformer-pytorch
