torch
rich
tqdm
wandb
transformers
datasets
sentencepiece
pytorch-lightning
scikit-learn
hydra-core
nltk
vllm