einops>=0.3
filelock
lie_learn
torch>=1.6
