torch
torchaudio
numpy
einops
loralib
vector-quantize-pytorch
pretty_midi
mido
music21
midi2audio
soundfile
fonttools
matplotlib
tensorboard
wandb
tqdm
google-api-python-client
google-auth-oauthlib
httplib2

[flash]
flash_attn
