rich>=13.0.0
torch>=2.1.0
torchaudio>=2.1.0
numpy>=1.24.0
scipy>=1.11.0
librosa>=0.10.0
soundfile>=0.12.0
pyyaml>=6.0
einops>=0.7.0
rotary-embedding-torch>=0.5.0
transformers>=4.35.0
huggingface-hub>=0.23.0
omegaconf>=2.2.0
ml-collections>=1.0.0
tqdm>=4.60.0
beartype>=0.14.0
spafe>=0.3.0

[cpu]
torch>=2.1.0
torchaudio>=2.1.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
ruff>=0.4.0
mypy>=1.0.0

[train]
wandb>=0.15.0
auraloss>=0.4.0
pytorch-lightning>=2.0.0
accelerate>=0.25.0
loralib>=0.1.0
