torch>=1.4
pytorch-lightning>=1.1.0
torchvision>=0.5
scikit-learn
gym