Skip to content

[WIP] Ray Data + Ray Train tutorial#3763

Draft
crypdick wants to merge 20 commits intopytorch:mainfrom
crypdick:ray-data-ray-train
Draft

[WIP] Ray Data + Ray Train tutorial#3763
crypdick wants to merge 20 commits intopytorch:mainfrom
crypdick:ray-data-ray-train

Conversation

@crypdick
Copy link
Contributor

@crypdick crypdick commented Feb 10, 2026

Adds a new tutorial showcasing streaming data loading using Ray Data with Ray Train. Will move out of draft once I get reviews from the Data and Train teams.

cc @pcmoritz @robertnishihara @matthewdeng @richardliaw @akshay-anyscale

@pytorch-bot
Copy link

pytorch-bot bot commented Feb 10, 2026

🔗 Helpful Links

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

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

❌ 1 New Failure, 1 Cancelled Job

As of commit d8b40fb with merge base 6e608db (image):

NEW FAILURE - The following job has failed:

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

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

@meta-cla meta-cla bot added the cla signed label Feb 10, 2026
# a high loss (~10–11).

###############################################################################
# Checkpointing
Copy link
Contributor Author

Choose a reason for hiding this comment

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

debating whether I should slim this section down, given that I'm not checkpointing in the tutorial

Choose a reason for hiding this comment

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

Let's remove this section and point to the checkpointing user guide in Ray Docs. That way we can direct people to new features and avoid showing outdated apis (ex: the TorchTrainer.restore() API is deprecated).

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Makes sense @justinvyu . I rewrote it without code, but kept the section because I wanted to discuss some of the nice checkpointing features. Lmk what you think.

@svekars svekars added the ray PRs related to tutorials that use the ray project: https://github.com/ray-project/ray label Feb 10, 2026
# a high loss (~10–11).

###############################################################################
# Checkpointing

Choose a reason for hiding this comment

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

Let's remove this section and point to the checkpointing user guide in Ray Docs. That way we can direct people to new features and avoid showing outdated apis (ex: the TorchTrainer.restore() API is deprecated).

@svekars svekars self-assigned this Feb 12, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cla signed ray PRs related to tutorials that use the ray project: https://github.com/ray-project/ray

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants