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/tensor/moe_tensor
Haze188 416580b314
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
5 months ago
..
__init__.py
api.py [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
moe_info.py