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 frequently cancelled, sometimes mid-checkout

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