torch==2.3.1
torchaudio==2.3.0+cu121
torchvision==0.18.0+cu121
transformers==4.43.1
trl==0.9.6
vllm @ https://github.com/vllm-project/vllm/releases/download/v0.5.3.post1/vllm-0.5.3.post1-cp38-cp38-manylinux1_x86_64.whl#sha256=9b9e17d5d36e330f12296fce66eea01508a28c0b37a1b279e033ef23157c97c2
vllm-flash-attn==2.5.9.post1
wandb==0.17.0
xformers==0.0.26.post1
flashinfer