requirements.txt

nltk>=3.7
numpy>=1.21.3
optuna>=2.10.0
pandas>=1.3.5
PyYAML>=6.0
scikit-learn>=1.0.1
scipy>= 1.7.1
seaborn>=0.11.2
SoundFile>=0.10.3.post1
librosa>=0.9.2
ir_datasets>=0.5.3

# --------- pytorch --------- #
torch>=1.12.1+cu116
torchaudio>=0.10.0+cu102
torchvision>=0.11.1+cu102
pytorch-lightning>=1.6.3
torchmetrics>=0.7.0
lightning-bolts>=0.5.0

# --------- transformers --------- #
transformers>=4.19.2
tokenizers>=0.12.1
datasets>=2.2.1
sentence-transformers>=2.2.0

# --------- hydra --------- #
hydra-core>=1.1.2
hydra-colorlog>=1.1.0
hydra-optuna-sweeper>=1.1.2

# --------- loggers --------- #
wandb>=0.12.9
# neptune-client
# mlflow
# comet-ml
tensorboard>=2.7.0

# --------- linters --------- #
pre-commit      # hooks for applying linters on commit
black           # code formatting
isort           # import sorting
flake8          # code analysis
nbstripout      # remove output from jupyter notebooks

# --------- others --------- #
python-dotenv   # loading env variables from .env file
rich            # beautiful text formatting in terminal
pytest          # tests
sh              # for running bash commands in some tests
pudb            # debugger