ColossalAI/colossalai/nn
Frank Lee e2089c5c15
adapted for sequence parallel (#163)
2022-01-20 13:44:51 +08:00
..
layer adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
loss Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
lr_scheduler Added MoE parallel (#127) 2022-01-07 15:08:36 +08:00
metric Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
model Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
optimizer refactor kernel (#142) 2022-01-13 16:47:17 +08:00
__init__.py Layer integration (#83) 2021-12-27 15:04:32 +08:00
init.py Layer integration (#83) 2021-12-27 15:04:32 +08:00