torch
tensorflow
tensorflow_datasets
pickle5>=0.0.11; python_version < "3.8" and python_version >= "3.6"
ray==1.6.0
datasets~=1.17
