Make functions visible to make a custom `fused_moe`

ext-torch/moe/__init__.py

@@ -2,7 +2,7 @@ import torch
 
 from ._ops import add_op_namespace_prefix, ops
 from .fused_marlin_moe import fused_marlin_moe
-from .fused_moe import fused_moe, fused_topk, grouped_topk
+from .fused_moe import fused_experts, fused_moe, fused_topk, grouped_topk
 from .scalar_type import ScalarType, scalar_types
 
 
@@ -80,7 +80,10 @@ def topk_softmax(
 __all__ = [
     "gptq_marlin_moe_repack",
     "awq_marlin_moe_repack",
+    "fused_experts",
     "fused_marlin_moe",
+    "fused_topk",
+    "grouped_topk",
     "moe_sum",
     "moe_align_block_size",
     "topk_softmax",