datasets
numpy
scipy
torchnet
transformers==4.21.0
smdebug
humanize