numpy
pandas
scikit-learn
matplotlib
seaborn
torch>=1.6.0
torchvision
torchtext==0.7
pytorch-nlp
transformers
joblib
tqdm
awscli
boto3
botocore
google-cloud-storage
protobuf==3.19.4
nltk
pyrouge
rouge
tensorboard
wandb
