tensorflow
tensorflow_datasets
torch
transformers==4.1.1
lit-nlp
gunicorn