Commit Graph

4 Commits

Author SHA1 Message Date
Hakjin Lee
46c009dba4 [format] Run lint on colossalai.engine (#3367) 2023-04-05 23:24:43 +08:00
Frank Lee
11f54c7b6b [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00
HELSON
84fd7c1d4d add moe context, moe utilities and refactor gradient handler (#455) 2022-03-18 16:38:32 +08:00
Frank Lee
e2089c5c15 adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00