torch>=1.9
ninja
