numpy
torch
torchaudio
einops
