modelscope / modelscope/ms-swift

npu-megatron

Open
#6,024 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

npu
Dominant language
Python
Stars
15.7k
Forks
1.7k
Avg merge
1d 16h
Merged PRs (30d)
136

Description

npu环境下,支持megatron分布训练吗?如果无法使用megatron,有没有其他tensor_parallel和pipeline_parallel的方法能够适用于npu训练?
感谢解答!

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

No files or tests are named. Start by locating the existing NPU and Megatron or distributed-training entry points, then determine whether tensor parallelism or pipeline parallelism is supported. Done should be a confirmed supported approach or a clearly documented limitation and alternative.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
distributed-systems, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.