torch>=2.6
triton
ninja

[bench]
transformers
accelerate
peft
diffusers
matplotlib
