torch==1.4.0
tensorflow==1.15.0
scikit-learn==0.22.1
stable-baselines==2.9.0
numpy==1.20.3
pykeops==1.5
geomloss==0.2.4
mpi4py==3.0.3
pygame==1.9.6
gym==0.15.4