chenxun.p 6682f5d92a fix reduce scatter async bug		2023-10-17 15:10:07 +08:00
..
__init__.py	feat(moe): add moe module (#182)	2023-09-27 15:54:53 +08:00
embedding.py	fix(model/embedding.py): ci lint check error (#345)	2023-09-21 14:46:22 +08:00
linear.py	impl reduce scatter async	2023-10-17 11:15:54 +08:00
loss.py	initial commit	2023-07-06 12:55:23 +08:00
metrics.py	feat(train): add fsdp training option (#293)	2023-10-09 18:59:31 +08:00
modeling_internlm.py	feat(model/linear.py): set block 0 full weight	2023-10-16 20:13:59 +08:00
modeling_moe.py	feat(moe): add moe module (#182)	2023-09-27 15:54:53 +08:00
moe.py	feat(moe): add moe module (#182)	2023-09-27 15:54:53 +08:00
multi_head_attention.py	feat(model/linear.py): set block 0 full weight	2023-10-16 20:13:59 +08:00
norm.py	Merge develop to main (#233)	2023-08-24 22:03:04 +08:00
utils.py	fix reduce scatter async bug	2023-10-17 15:10:07 +08:00