torch>=1.5.1
ninja
