numpy>=1.18.0
scikit-learn>=0.22.0
torch>=1.7.0

# optional
black # for formatting
matplotlib # for plotting