PyYAML>=5.1
numpy
pandas
torch