torch
numpy
librosa
scipy==1.4.1
tensorboard==2.0
soundfile
matplotlib==3.1.3
nnaudio
ring-attention-pytorch
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/triton-nightly