[Data] [Docs] Docs for working with PyTorch (ray-project#36880)

Adds a guide for how PyTorch users can use Ray Data. This includes a migration guide from Torch Dataset/DataLoader to Ray Data. Closes ray-project#31418 --------- Signed-off-by: amogkam <[email protected]>
Mark2000 · Jun 30, 2023 · 80e458c · 80e458c
1 parent eb07a52
commit 80e458c
Show file tree

Hide file tree

Showing 6 changed files with 511 additions and 3 deletions.
diff --git a/doc/source/data/api/from_other_data_libs.rst b/doc/source/data/api/from_other_data_libs.rst
@@ -86,3 +86,9 @@ For PyArrow Users
      - :meth:`ds.groupby() <ray.data.Dataset.groupby>`
    * - pa.Table.sort_by()
      - :meth:`ds.sort() <ray.data.Dataset.sort>`
+
+
+For PyTorch Dataset & DataLoader Users
+--------------------------------------
+
+For more details, see the :ref:`Migrating from PyTorch to Ray Data <migrate_pytorch>`.
diff --git a/doc/source/data/user-guide.rst b/doc/source/data/user-guide.rst
@@ -20,5 +20,7 @@ show you how achieve several tasks.
     working-with-images
     working-with-text
     working-with-tensors
+    working-with-pytorch
     batch_inference
+    data-internals
     performance-tips