albumentations
einops
torch>=1.8.0
wandb
