ColossalAI/colossalai/moe at 8d3d7f3cbdb68cab7878abcab6964dff6ebc65f5 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-17 19:31:14 +00:00

Files

History

botbw 335ad3c6fb [moe] implement tp

2024-07-19 07:30:17 +00:00

..

__init__.py

[MoE/ZeRO] Moe refactor with zero refactor (#5821 )

2024-06-28 14:00:08 +08:00

_operation.py

[moe] implement tp

2024-07-19 07:30:17 +00:00

load_balance.py

[MoE/ZeRO] Moe refactor with zero refactor (#5821 )

2024-06-28 14:00:08 +08:00

manager.py

fix some typo (#5307 )

2024-01-25 13:56:27 +08:00

utils.py

[MoE/ZeRO] Moe refactor with zero refactor (#5821 )

2024-06-28 14:00:08 +08:00