-r requirements_common.txt

torch==2.6.0.*
torchvision==0.21.0.*
torchaudio==2.6.0.*

flash-attn

transformer-engine[pytorch]==2.2.0
deepspeed==0.16.4
vllm==0.8.4
