area/examplesgood first issuehelp wanted
Repository metrics
- Stars
- (2,112 stars)
- PR merge metrics
- (PR metrics pending)
Description
It would be good to run GPU E2E tests for DeepSpeed and MLX example Notebooks to tests our MPI-based runtimes.
- https://github.com/kubeflow/trainer/blob/master/examples/mlx/language-modeling/fine-tune-llama.ipynb
- https://github.com/kubeflow/trainer/blob/master/examples/deepspeed/text-summarization/T5-Fine-Tuning.ipynb
That should help us detect MPI errors earlier, like here: https://github.com/kubeflow/trainer/pull/3748
/good-first-issue /area examples