hydra-core
torch
torchvision
gradio
einops