Skip to content

Add MoE + expert-parallel Python API surface, PyTorch reference, and tests#389

Open
Anerudhan wants to merge 3 commits into
NVIDIA:developfrom
Anerudhan:moe_ep_api
Open

Add MoE + expert-parallel Python API surface, PyTorch reference, and tests#389
Anerudhan wants to merge 3 commits into
NVIDIA:developfrom
Anerudhan:moe_ep_api

Document and stub the MoeEp.backward call contract

50df079
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs