torch>=2.1.0
transformers>=4.36.0
numpy

[audio]
torchaudio>=2.1.0
librosa>=0.10.0

[dev]
pytest
pytest-benchmark
