torch>=2.1
ninja
