NVIDIA / NVIDIA/Megatron-LM

[QUESTION] Support for Heterogeneous Parallelism in Multimodal Training

Open
#1,375 4 comments 0 reactions 1 assignee Claimed by @Phlip79 View on GitHub
community-request enhancement waiting-on-maintainers
Dominant language
Python
Stars
17.9k
Forks
4.5k
Avg merge
4d 6h
Merged PRs (30d)
271

Description

I have been using MegatronLM to train multimodal models and successfully followed the example under examples/multimodal. However, for efficient training, multimodal models often require different parallelism strategies for each component, as vision models are typically smaller than the LLM in such setups.

**Does MegatronLM support heterogeneous parallelism strategies**, where different models within a multimodal system can use distinct parallelization techniques? If not, are there any recommended workarounds?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.