torch
torchaudio
einops
stable-audio-tools