torch>=2.0
torchvision
einops
matplotlib
numpy
requests
pillow
