# Public CUDA 13.4 prerelease PyTorch nightly pins for Rubin bring-up.
# Installed when INSTALL_RUBIN_PRERELEASE=true (see docker/Dockerfile).
# TODO(rubin-bringup): delete once Rubin APIs ship in regular PyTorch wheels
# (vllm-project/vllm#49735).
--extra-index-url https://download.pytorch.org/whl/nightly/cu134
--extra-index-url https://flashinfer.ai/whl/cu134
--extra-index-url https://flashinfer.ai/whl

# Match each architecture's torch pin to the latest torchvision wheel.
torch==2.15.0.dev20260909+cu134; platform_machine == "x86_64"
torch==2.15.0.dev20260908+cu134; platform_machine == "aarch64"
torchvision==0.30.0.dev20260909+cu134
torchaudio==2.11.0.dev20260911+cu134

# docker/Dockerfile strips these packages from requirements/cuda.txt whenever
# INSTALL_RUBIN_PRERELEASE=true, so they must be pinned here instead to
# still get installed.
nvidia-cutlass-dsl[cu13]==4.8.0.dev0
flashinfer-python==0.6.18.post1
quack-kernels==0.6.5

cuda-python==13.4.1
cuda-bindings==13.4.1
flashinfer-jit-cache==0.6.18.post1+cu134
flashinfer-cubin==0.6.18.post1
nixl==1.4.1
