einops
torch>=1.6
kornia>=0.4.0
