numpy==1.19.4
ray==1.2.0
pytest==6.2.3
matplotlib==3.3.3
torchvision==0.8.2
PyYAML==5.4.1
scikit_learn==0.24.1
