setuptools
pyyaml
torch<2.0,>=1.10
torchaudio
torchlibrosa
pytorch-lightning<2.0,>=1.7
nltk
spacy
tensorboard
transformers
omegaconf

[dev]
pytest
flake8
black
ipykernel

[train]
torchvision
torchtext
torchmetrics
pip
hydra-core
h5py
sentencepiece
py7zr
jupyter
flake8
pytest
bokeh
sentence-transformers
numpy
scikit-image
gensim
click
pytz
pyemd
pydantic>=1.8.2
matplotlib
scikit-learn-intelex
audiomentations
bert-score
black
torch-optimizer
torchopenl3
hydra-optuna-sweeper
hydra-colorlog
timm
language_tool_python
hydra_colorlog
aac-datasets~=0.4.0
aac-metrics~=0.5.3
deepspeed==0.9.5
intel_extension_for_pytorch
