torch>=1.8.1
numpy
pandas
tqdm
tabulate
pytest
scikit-learn