uv-torch

crates/uv-torch/ is the special-case integration for the PyTorch wheel index: support for --torch-backend and the PyTorch-flavored index URLs (https://download.pytorch.org/whl/cu121 etc.).

Purpose

PyTorch publishes per-CUDA wheels at variant URLs. uv users want a turnkey way to ask for "the right torch for this machine," and that's exactly what this crate (and the --torch-backend flag) provides.

Key abstractions

• TorchBackend — Cpu, Cuda(version), Rocm(version), Metal, Auto.
• Helpers that map a backend choice to an index URL and recognize a system's available accelerators when Auto.

See also

• uv-installer, uv-resolver — consumers.