einops
torch>=2.0.0
torchaudio>=2.0.0
