From 20ab80ad8f035d99ef0257476bf6e3819b14ea3b Mon Sep 17 00:00:00 2001
From: Matthew Woolf <48444393+mjwoolf@users.noreply.github.com>
Date: Wed, 3 Dec 2025 06:47:57 -0800
Subject: [PATCH] Update pytorch.md to include information about supporting
 CUDA 13.0.x (#16957)

## Summary

This change updates the guide about integration with pytorch to include
the CUDA 13.0 variant.

## Test Plan

I generated the documentation and verified that it looked correct.
---
 docs/guides/integration/pytorch.md | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/docs/guides/integration/pytorch.md b/docs/guides/integration/pytorch.md
index 5bbb6bb94..94b111083 100644
--- a/docs/guides/integration/pytorch.md
+++ b/docs/guides/integration/pytorch.md
@@ -103,6 +103,15 @@ In such cases, the first step is to add the relevant PyTorch index to your `pypr
     explicit = true
     ```
 
+=== "CUDA 13.0"
+
+    ```toml
+    [[tool.uv.index]]
+    name = "pytorch-cu130"
+    url = "https://download.pytorch.org/whl/cu130"
+    explicit = true
+    ```
+
 === "ROCm6"
 
     ```toml
@@ -184,6 +193,21 @@ Next, update the `pyproject.toml` to point `torch` and `torchvision` to the desi
     ]
     ```
 
+=== "CUDA 13.0"
+
+    PyTorch doesn't publish CUDA builds for macOS. As such, we gate on `sys_platform` to instruct uv to limit
+    the PyTorch index to Linux and Windows, falling back to PyPI on macOS:
+
+    ```toml
+    [tool.uv.sources]
+    torch = [
+      { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+    ]
+    torchvision = [
+      { index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+    ]
+    ```
+
 === "ROCm6"
 
     PyTorch doesn't publish ROCm6 builds for macOS or Windows. As such, we gate on `sys_platform` to instruct uv