lightning<3.0,>=2.4
torch<3.0,>=2.6
torchaudio<3.0,>=2.6
gin-config<1.0,>=0.5
einops<1.0,>=0.8
transformers<5.0,>=4.50
sentence_transformers<5.2,>=5.0
omar-rq~=0.2
huggingface_hub

[dev]
amclap[eval,test,train]

[eval]
scikit_learn>=1.5
scipy>=1.14
pandas
matplotlib
essentia
cffi

[llm_preprocess]
vllm>=0.5

[test]
pytest
ruff~=0.15
pandas

[train]
wandb
pandas
datasets>=3.0
