[SPARK-44264] DeepSpeed Distributor

• Type: Improvement
• Status: Resolved
• Priority: Critical
• Resolution: Fixed
• Affects Version/s: 3.4.1
• Fix Version/s: 3.5.0
• Component/s: ML, PySpark
• Labels: pull-request-available
• Assignee: Unassigned
• Reporter: Lu Wang
• Shepherd: Hyukjin Kwon
• Created: 30/Jun/23 20:01
• Updated: 02/May/24 06:56
• Resolved: 11/Jul/23 22:37

Description

To make it easier for Pyspark users to run distributed training and inference with DeepSpeed on spark clusters using PySpark. This was a project determined by the Databricks ML Training Team.

Issue Links

• links to: [Github] Pull Request #42067 (mathewjacob1002)
• links to: [Github] Pull Request #42118 (mathewjacob1002)
• links to: GitHub Pull Request #41946