tensorflow==2.3.1
torch>=1,<2
ray>=1.0
transformers>=3.5.1
dask[complete]>=2.30
tensorflow_datasets
ray==1.2.0