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 1 month ago
..
core Replace import 1 month ago
data Replace import 1 month ago
fp16_deprecated Source code to initialize megatron 1 month ago
fused_kernels Replace import 1 month ago
model Replace import 1 month ago
mpu/tests Replace import 1 month ago
static Source code to initialize megatron 1 month ago
text_generation Replace import 1 month ago
tokenizer Source code to initialize megatron 1 month ago
__init__.py Replace import 1 month ago
arguments.py Replace import 1 month ago
checkpointing.py Replace import 1 month ago
dist_signal_handler.py Replace import 1 month ago
global_vars.py Replace import 1 month ago
indexer.py Replace import 1 month ago
initialize.py Replace import 1 month ago
log_handler.py Source code to initialize megatron 1 month ago
memory.py Replace import 1 month ago
microbatches.py Source code to initialize megatron 1 month ago
optimizer_param_scheduler.py Source code to initialize megatron 1 month ago
text_generation_server.py Replace import 1 month ago
theoretical_memory_usage.py Source code to initialize megatron 1 month ago
training.py Replace import 1 month ago
utils.py Replace import 1 month ago

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

Python Markdown Shell C++ other

Contributors (2)