ColossalAI/colossalai/context
HELSON f24b5ed201
[MOE] remove old MoE legacy (#493)
2022-03-22 17:37:16 +08:00
..
process_group_initializer [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469) 2022-03-21 13:35:04 +08:00
random add moe context, moe utilities and refactor gradient handler (#455) 2022-03-18 16:38:32 +08:00
__init__.py add moe context, moe utilities and refactor gradient handler (#455) 2022-03-18 16:38:32 +08:00
config.py update default logger (#100) (#101) 2022-01-04 20:03:26 +08:00
moe_context.py [format] polish name format for MOE (#481) 2022-03-21 23:19:47 +08:00
parallel_context.py add moe context, moe utilities and refactor gradient handler (#455) 2022-03-18 16:38:32 +08:00
parallel_mode.py [MOE] remove old MoE legacy (#493) 2022-03-22 17:37:16 +08:00