zhanglei 92a31732f9 fix the moe_loss_coeff bug		2023-08-15 18:47:19 +08:00
..
communication	refactor(scheduler): rewrite pipeline scheduler (#138)	2023-08-03 11:48:12 +08:00
context	add expert data support and fix bugs	2023-08-10 16:07:35 +08:00
scheduler	fix the moe_loss_coeff bug	2023-08-15 18:47:19 +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	refactor(scheduler): rewrite pipeline scheduler (#138)	2023-08-03 11:48:12 +08:00
trainer.py	feat(core/scheduler): support pipeline parallel (#98)	2023-07-24 20:52:09 +08:00