pytest~=8.3.1
triton~=3.0.0
torch~=2.2.1
numpy==1.26.4
