Wei ZhaoandWei Zhao 2427094152 [Feature] Support EPLB for DeepSeek v4 Mega Moe (#43339) ...

Signed-off-by: wzhao18 <wzhao18.sz@gmail.com>
Co-authored-by: Wei Zhao (Engrg-Hardware 1) <weizha@login-lyris01.lyris.clusters.nvidia.com>

2026-06-02 10:56:44 -07:00

..

policy

[EPLB] Simplify EPLB rearrange by only returning one map (#36267)

2026-03-18 20:34:00 -04:00

__init__.py

Abstract eplb algo (#26471)

2025-12-04 19:09:09 +00:00

async_worker.py

[EPLB] Remove asyncio infrastructure from Async EPLB (#40730)

2026-04-24 00:21:15 +00:00

eplb_communicator.py

[Refactor] Nixl util using lazy init (#41392)

2026-05-09 17:46:52 -04:00

eplb_state.py

[MoE Refactor] EPLB refactoring for FusedMoE (#41055)

2026-05-12 14:16:31 -04:00

eplb_utils.py

[Feature] Support EPLB for DeepSeek v4 Mega Moe (#43339)

2026-06-02 10:56:44 -07:00

rebalance_execute.py

[RFC][EPLB][#32028] Remove dead torch.accelerator.synchronize() from sync path (#40733)

2026-05-22 15:19:24 -04:00