requirements.txt

# --------- pytorch --------- #
torch>=1.13.0
torchvision>=0.11.0
pytorch-lightning==2.0.0
torchmetrics~=0.11.0

# --------- hydra --------- #
hydra-core==1.3.1
hydra-colorlog==1.2.0
hydra-optuna-sweeper==1.2.0

# --------- loggers --------- #
wandb~=0.13.11
# neptune-client
# mlflow
# comet-ml

# --------- others --------- #
pyrootutils~=1.0.4
pre-commit      # hooks for applying linters on commit
rich~=13.3.1
pytest~=7.1.3
# sh            # for running bash commands in some tests (linux/macos only)

Hydra~=2.5
omegaconf~=2.2.3
transformers~=4.20.0
einops~=0.5.0
pandas~=1.5.2
wandb~=0.13.9
nltk~=3.8.1
dask~=2021.3.0
tqdm~=4.64.1
paramiko~=2.12.0
requests~=2.28.2
scp~=0.14.4
packaging~=23.0
setuptools~=67.0.0

numpy~=1.24.4
orjson~=3.9.2
gensim~=4.3.1
pyterrier~=0.1.5
beir~=2.0.0
scikit-learn~=1.3.0
langchain~=0.0.240
matplotlib~=3.7.2
openai~=0.27.8
typer~=0.9.0
python-dotenv~=1.0.0
wget~=3.2
pyserini~=0.21.0
seaborn~=0.13.0