Skip to content

Modernize the Ray Tune tutorial#3694

Merged
svekars merged 29 commits intopytorch:mainfrom
crypdick:ray-version-bump
Jan 8, 2026
Merged

Modernize the Ray Tune tutorial#3694
svekars merged 29 commits intopytorch:mainfrom
crypdick:ray-version-bump

Conversation

@crypdick
Copy link
Contributor

@crypdick crypdick commented Dec 11, 2025 •
edited by pytorch-bot bot
Loading

This is the first of several PRs against the Pytorch Tutorials. This PR unstales the Ray Tune tutorial:

  • bumps Ray version from 2.7.2 -> 2.52.1
  • update the tutorial with the latest code from the Pytorch CIFAR10 tutorial which it is based off of
  • update to use the latest Ray Tune APIs
  • fixed the norm values for CIFAR10
  • rewrote most of the prose
  • added new sections for sampling, observability, conclusion, additional links
  • updated tags in the doc indices. Will need approval on the new Ray-Distributed tag.

Opening this as a draft PR so that I can get feedback from the Ray Tune core devs first

cc @subramen @albanD @jbschlosser @pcmoritz @robertnishihara @matthewdeng @richardliaw

@pytorch-bot
Copy link

pytorch-bot bot commented Dec 11, 2025 •
edited
Loading

🔗 Helpful Links

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

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

✅ No Failures

As of commit 7dd199d with merge base 00d8b24 (image):
💚 Looks good so far! There are no failures yet. 💚

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

@meta-cla meta-cla bot added the cla signed label Dec 11, 2025
crypdick
crypdick commented Dec 11, 2025
View reviewed changes
crypdick
crypdick commented Dec 11, 2025
View reviewed changes
@svekars svekars added core Tutorials of any level of difficulty related to the core pytorch functionality intro labels Dec 11, 2025
@svekars svekars added the ray PRs related to tutorials that use the ray project: https://github.com/ray-project/ray label Dec 15, 2025
@crypdick
Copy link
Contributor Author

crypdick commented Dec 15, 2025

cc @xinyuangui2

xinyuangui2
xinyuangui2 reviewed Dec 15, 2025
View reviewed changes
xinyuangui2
xinyuangui2 approved these changes Dec 16, 2025
View reviewed changes
Copy link

@xinyuangui2 xinyuangui2 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good from ray tune side.

@crypdick crypdick changed the title [WIP] Modernize the Ray Tune tutorial Modernize the Ray Tune tutorial Dec 16, 2025
@crypdick crypdick marked this pull request as ready for review December 16, 2025 23:23
@crypdick
Copy link
Contributor Author

crypdick commented Dec 16, 2025

@svekars Ready for review from the PyTorch side :)

@crypdick crypdick mentioned this pull request Dec 18, 2025
Open
@crypdick
Copy link
Contributor Author

crypdick commented Jan 5, 2026

Happy new year @svekars . Gentle bump on the review on this PR. This one is soft blocking other contributions because of the Ray version bump.

@svekars
Copy link
Contributor

svekars commented Jan 5, 2026

Thanks for the reminder! Will review today.

svekars
svekars reviewed Jan 6, 2026
View reviewed changes
svekars
svekars reviewed Jan 6, 2026
View reviewed changes
Copy link
Contributor

@svekars svekars left a comment •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I also see that now it takes 32 minutes to build this tutorial while the current version takes 12 minutes. Can we reduce the time of the build? We could try to put on a different worker and see if it helps.

@crypdick
Copy link
Contributor Author

crypdick commented Jan 6, 2026

@svekars I made the following changes to speed up the tutorial:

  • I reverted the number of data loader workers from 2 back to 8
  • I made a GPU available to the tutorial by changing the CI worker to linux.g5.4xlarge.nvidia.gpu

svekars
svekars reviewed Jan 7, 2026
View reviewed changes
svekars
svekars reviewed Jan 7, 2026
View reviewed changes
svekars
svekars approved these changes Jan 7, 2026
View reviewed changes
Copy link
Contributor

@svekars svekars left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM - I added a couple of very minor formatting comments! Thank you for adjusting the execution time and updating the tutorial!

@crypdick @svekars
Apply suggestions from code review
7dd199d 
Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
@svekars svekars merged commit ebe3e9f into pytorch:main Jan 8, 2026
38 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@svekars svekars svekars approved these changes

+1 more reviewer

@xinyuangui2 xinyuangui2 xinyuangui2 approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

cla signed core Tutorials of any level of difficulty related to the core pytorch functionality intro ray PRs related to tutorials that use the ray project: https://github.com/ray-project/ray

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@crypdick @svekars @xinyuangui2