# Use the PyTorch ROCm wheel index (choose the stream that matches your system)
--extra-index-url https://download.pytorch.org/whl/rocm6.3

pre-commit
torch==2.8.0+rocm6.3
torchaudio==2.8.0+rocm6.3
torchvision==0.23.0+rocm6.3
transformers==4.57.1
qwen-vl-utils==0.0.11
datasets
setuptools
tqdm
wandb
psutil
numpy
accelerate
pydantic
sglang[all]==0.5.4
openai-harmony
tensorboard