requirements.txt

# Core Deep Learning Framework
torch==2.4.0
torchaudio==2.4.0
lightning==2.5.1.post0

# ML/DL Libraries
transformers==4.51.1
accelerate==1.5.2
datasets==3.6.0
tokenizers==0.21.1
safetensors==0.5.3

# Scientific Computing
numpy==1.25.0
scipy==1.15.2
scikit-learn==1.6.1
pandas==2.2.3

# Audio Processing
librosa==0.11.0
audioread==3.0.1
soundfile==0.13.1
pesq==0.0.4
auraloss==0.4.0
nnAudio==0.3.3
julius==0.2.7
soxr==0.5.0.post1
mir_eval==0.8.2
jams==0.3.4
msaf==0.1.80

# Visualization & Monitoring
matplotlib==3.10.1
seaborn==0.13.2
tensorboard==2.19.0
wandb==0.19.8
gpustat==1.1.1

# Configuration & CLI
hydra-core==1.3.2
omegaconf==2.3.0
fire==0.7.1
click==8.1.8

# Deep Learning Utils
einops==0.8.1
einx==0.3.0
x-transformers==2.4.14
x-clip==0.14.4
ema-pytorch==0.7.7
schedulefree==1.4.1
torchmetrics==1.7.1

# Data Processing
h5py==3.13.0
pyarrow==19.0.1
pillow==11.1.0

# Text Processing
ftfy==6.3.1
regex==2024.11.6
pypinyin==0.54.0
textgrid==1.6.1
pylrc==0.1.2

# Model Management
modelscope==1.27.1

# Utilities
tqdm==4.67.1
loguru==0.7.3
joblib==1.4.2
easydict==1.13
addict==2.4.0
beartype==0.21.0

# Others
triton==3.0.0
muq==0.1.0
vmo==0.30.5

# others
gradio
einops
beartype

huggingface-hub==0.33.5