torch>=1.6
torchaudio
data-science-types>=0.2
einops>=0.6
einops-exts>=0.0.3
