accelerate
audioread
einops
jukebox
librosa==0.9.1
matplotlib
numpy
packaging
pip
prefigure
resampy
scipy
torch
torchaudio
torchvision
tqdm
wandb

[dev]
