Torch Function modes x torch.compile tutorial

[mlazos]

Fixes #ISSUE_NUMBER

Description

Checklist

• The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• Only one issue is addressed in this pull request
• Labels from the issue that this PR is fixing are added to this pull request
• No unnecessary issues are included into this pull request.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/3320

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• [Infra] Jobs got intermittently cancelled/fail midway checkout

❌ 1 Cancelled Job, 3 Unrelated Failures

As of commit 4e854d5 with merge base 752e0c9 ():

CANCELLED JOB - The following job was cancelled. Please retry:

• Build tutorials / pytorch_tutorial_build_worker (14, 15, linux.g5.4xlarge.nvidia.gpu) (gh)

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

• Build tutorials / pytorch_tutorial_build_worker (10, 15, linux.g5.4xlarge.nvidia.gpu) (gh) (matched linux rule in flaky-rules.json)
  The runner has received a shutdown signal. This can happen when the runner service is stopped, or a manually started runner is canceled.
• Build tutorials / pytorch_tutorial_build_worker (7, 15, linux.g5.4xlarge.nvidia.gpu) (gh) (matched linux rule in flaky-rules.json)
  The runner has received a shutdown signal. This can happen when the runner service is stopped, or a manually started runner is canceled.
• Build tutorials / pytorch_tutorial_build_worker (9, 15, linux.g5.4xlarge.nvidia.gpu) (gh) (matched linux rule in flaky-rules.json)
  The runner has received a shutdown signal. This can happen when the runner service is stopped, or a manually started runner is canceled.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[svekars]

Looks good, added some minor suggestions.

[williamwen42]

Content lgtm

[svekars]

it seems to be failing: https://github.com/pytorch/tutorials/actions/runs/14373299892/job/40300235591?pr=3320#step:9:4837
Can you try to put on a linux.g5.4xlarge.nvidia.gpu in this file: https://github.com/pytorch/tutorials/blob/main/.jenkins/metadata.json