torch>=1.10
triton==2.0.0.dev20220323
numpy
pytest
