torch==2.0.1
torchvision==0.13.1
kornia==0.6.10
numpy==1.23.5
matplotlib==3.5.1
einops==0.4.1
