sample/jetengine_ext/kernels/__init__.py

from .triton.fused_moe import fused_moe  # re-export

__all__ = ["fused_moe"]