torch>=1.5
ninja
