einops>=0.4
einops-exts
rotary-embedding-torch
torch>=1.10
torchvision
tqdm
