You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
Eric_lai 3839035126 Replace import 2 weeks ago
docs Source code to initialize megatron 2 weeks ago
examples Replace import 2 weeks ago
images Source code to initialize megatron 2 weeks ago
megatron Replace import 2 weeks ago
tasks Replace import 2 weeks ago
tests Replace import 2 weeks ago
tools Replace import 2 weeks ago
.coveragerc Source code to initialize megatron 2 weeks ago
.gitignore Replace import 2 weeks ago
.gitlab-ci.yml Source code to initialize megatron 2 weeks ago
CODEOWNERS Source code to initialize megatron 2 weeks ago
CONTRIBUTING.md Source code to initialize megatron 2 weeks ago
Dockerfile.ci Source code to initialize megatron 2 weeks ago
LICENSE Source code to initialize megatron 2 weeks ago
MANIFEST.in Source code to initialize megatron 2 weeks ago
README.md Initial commit 2 weeks ago
README_megatron.md Source code to initialize megatron 2 weeks ago
jet-tests.yml Source code to initialize megatron 2 weeks ago
pretrain_bert.py Replace import 2 weeks ago
pretrain_gpt.py Replace import 2 weeks ago
pretrain_ict.py Replace import 2 weeks ago
pretrain_retro.py Replace import 2 weeks ago
pretrain_t5.py Replace import 2 weeks ago
pretrain_vision_classify.py Replace import 2 weeks ago
pretrain_vision_dino.py Replace import 2 weeks ago
pretrain_vision_inpaint.py Replace import 2 weeks ago
pyproject.toml Source code to initialize megatron 2 weeks ago
report_theoretical_memory.py Source code to initialize megatron 2 weeks ago
setup.py Source code to initialize megatron 2 weeks ago

基于MindTorch的Megatron-LM,支持NPU运行

Python Markdown Shell C++ other

Contributors (2)