scikit-learn>=0.22.0
matplotlib>=2.2.3

[all]
torch==1.10.0+cu102
ninja>=1.10.2.2
numba>=0.53.1

[numba]
numba>=0.53.1

[torch-cpu]
torch>=1.8.0
ninja>=1.10.2.2

[torch-gpu]
torch==1.10.0+cu102
ninja>=1.10.2.2
