--extra-index-url https://download.pytorch.org/whl/cu126
torch==2.6.0+cu126
nvidia-nccl-cu12==2.21.5
triton==3.2.0
numpy==1.26.4
