You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/moe
Haze188 416580b314
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
5 months ago
..
__init__.py [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
_operation.py [moe] support mixtral (#5309) 10 months ago
load_balance.py [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
manager.py fix some typo (#5307) 10 months ago
utils.py [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago