-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
sglang[srt,torch-memory-saver]==0.4.6.post4
transformers==4.51.1
cuda-bindings==12.9.0
