InternLM/internlm/core
yingtongxiong 985465c96a merge upstream 2023-10-25 14:46:45 +08:00
..
communication Merge develop to main (#233) 2023-08-24 22:03:04 +08:00
context fix(*)/all-reduce for norm in sequence parallel (#443) 2023-10-25 14:16:32 +08:00
scheduler merge upstream 2023-10-25 14:46:45 +08:00
__init__.py initial commit 2023-07-06 12:55:23 +08:00
engine.py initial commit 2023-07-06 12:55:23 +08:00
gradient_handler.py initial commit 2023-07-06 12:55:23 +08:00
naive_amp.py feat(moe): add moe module (#182) 2023-09-27 15:54:53 +08:00
trainer.py feat(moe): add moe module (#182) 2023-09-27 15:54:53 +08:00