numpy>=1.20
matplotlib>=3.4.0
torch>=1.8.0
scipy
ninja
