Agent skill

Fused Moe Kernel

by ZJLi2013 in ZJLi2013/awesome-kernel-skills

Optimize Fused Mixture-of-Experts (MoE) kernels in Triton for NVIDIA and AMD GPUs.

No licenceAuto-check passed

Install Fused Moe Kernel

skills CLI
$ npx skills add ZJLi2013/awesome-kernel-skills --skill fused-moe-kernel -a claude-code

Project install by default; add -g for ~/.claude/skills/.

GitHub CLI
$ gh skill install ZJLi2013/awesome-kernel-skills fused-moe-kernel --agent claude-code

Project scope by default; add --scope user for a personal install. Needs GitHub CLI 2.90.0 or later (public preview).

Manual copy
$ git clone --depth 1 https://github.com/ZJLi2013/awesome-kernel-skills.git skills-src && mkdir -p .claude/skills && cp -r skills-src/skills/kernels/fused-moe .claude/skills/fused-moe-kernel && rm -rf skills-src

Use ~/.claude/skills/ instead of .claude/skills for a personal install. The folder must contain SKILL.md.

Claude Code skills documentation · loads skills from .claude/skills/

Facts

Skill name
fused-moe-kernel
GitHub stars
102
Token cost
~699 tokens
SKILL.md length
282 words
Files
3
Skills in repo
12
Repo updated
First seen
Licence
None found

At a glance

Optimize Fused Mixture-of-Experts (MoE) kernels in Triton for NVIDIA and AMD GPUs.

  • Works in 3 steps: Top-k gating: select k experts per token… → Sort tokens by assigned expert → Pad each expert's token count to be…
  • Optimizing MoE layers
  • SKILL.md covers Overview, Core Technique, Key Parameters and Verification, plus 2 more sections
  • Runs Python scripts from its folder; calls python

What it does

Fused Moe Kernel is an agent skill from ZJLi2013/awesome-kernel-skills. Optimize Fused Mixture-of-Experts (MoE) kernels in Triton for NVIDIA and AMD GPUs. Covers token-expert routing, grouped GEMM, cache-aware scheduling, and top-k gating fusion. Use when writing or optimizing MoE layers, expert routing, or grouped matrix multiplication.

Its SKILL.md is about 700 tokens, which your agent loads only when the skill is triggered. The skill folder holds 2 other files (for example `test_fused_moe.py` and `triton_template.py`).

It works with NVIDIA AI Platform.

When your agent uses it

  • Optimizing MoE layers
  • Grouped matrix multiplication

Example prompts

  • “/fused-moe-kernel”

Requirements

  • Python 3

Workflow steps

3 steps, taken from the first numbered list in SKILL.md.

  1. Top-k gating: select k experts per token (typically k=2)
  2. Sort tokens by assigned expert
  3. Pad each expert's token count to be divisible by BLOCK_SIZE_M

What it can do on your machine

Read from SKILL.md and the folder at commit aba7662. It shows what the files ask for, not the result of running them.

  • Tool permissions

    Pre-approves nothing: there is no allowed-tools line, so your agent's usual permission prompts apply.

    From allowed-tools in the SKILL.md frontmatter.

  • Runs code

    Ships script files (Python), which the agent can run.

    Shell commands in SKILL.md call:

    • python

    From the folder's file list and the shell code blocks in SKILL.md.

  • Network

    Links to these hosts (documentation or services it may open):

    • pytorch.org
    • github.com
    • emergentmind.com

    From URLs in SKILL.md, links to its own repository left out.

  • Credentials

    Names no API keys, tokens, secrets or passwords.

    From names ending in _API_KEY, _TOKEN, _SECRET, _KEY or _PASSWORD in SKILL.md.

Context cost

Fused Moe Kernel loads about 699 tokens when it runs. Until then it costs about 71 tokens; SKILL.md has 282 words of instructions outside code blocks.

Always · name and description, kept in context so the agent knows when to use it
~71
When it runs · the whole SKILL.md, loaded when a task matches
~699

Estimates: characters ÷ 4, the usual rule of thumb; real counts depend on the model's tokenizer. Scripts and assets cost tokens only if the agent reads them.

Safety

Auto-check passed

The automated check found no risky patterns in SKILL.md.

Automated static check — not a guarantee. Review scripts before installing. It scans the text of SKILL.md for risky patterns (piping downloads into a shell, reading credential files, hidden Unicode, destructive commands); files beside SKILL.md are not scanned.

SKILL.md

Without a licence we can't republish the file, so here is its outline and opening line. It has 282 words (~699 tokens).

“Fused MoE combines token routing, expert GEMM, and output accumulation into a single kernel, avoiding expensive scatter/gather operations in global memory.”

— opening of SKILL.md by ZJLi2013
name
fused-moe-kernel

Read the full SKILL.md on GitHub

Files

SKILL.md and 2 other files in skills/kernels/fused-moe of ZJLi2013/awesome-kernel-skills.

  • SKILL.md
  • test_fused_moe.py
  • triton_template.py

Open the folder on GitHubat commit aba7662

Compare with similar skills

Fused Moe Kernel next to the 5 skills that share the most tags, products or categories with it. Stars are the repository's; “used in” counts other GitHub owners with a copy.

Fused Moe Kernel compared with similar skills
SkillStarsUsed inTokensAuto-checkLicenceRepo updated
Fused Moe Kernel this skillZJLi2013/awesome-kernel-skills102—~699Automated safety check: PassNone
LLM Torch Profiler Analysissgl-project/sglang37k2 repos~6.4kAutomated safety check: PassApache-2.0
Skill InspectorNVIDIA/SkillSpector20k—~1.8kAutomated safety check: PassApache-2.0
Megatron-LM Container and Dependency SetupNVIDIA/Megatron-LM18k—~2.6kAutomated safety check: PassApache-2.0
Embeddings via 9Routerdecolua/9router30k—~604Automated safety check: PassMIT
Megatron-LM Base Image BumpNVIDIA/Megatron-LM18k—~2.8kAutomated safety check: PassApache-2.0

Similar skills

  • LLM Torch Profiler Analysis

    sgl-project/sglang

    Unified LLM torch-profiler triage skill for sglang, vllm, TensorRT-LLM, and TokenSpeed.

    37k GitHub starsUsed in 2 repos~6.4k tokens
    DevelopmentAuto-check passed
  • Skill Inspector

    NVIDIA/SkillSpector

    Official

    Decides whether an agent skill is safe to install by combining a SkillSpector static scan with the agent's own source review, ending in APPROVE, CAUTION or REJECT.

    20k GitHub stars~1.8k tokensUpdated today
    SecurityAuto-check passed
  • Official

    Walks an agent through working inside the Megatron-LM CI container and changing dependencies with uv, so lock files resolve the same locally and in CI.

    18k GitHub stars~2.6k tokensUpdated today
    DevelopmentAuto-check passed
  • Embeddings via 9Router

    decolua/9router

    Generates vector embeddings through the 9Router /v1/embeddings endpoint, using models from providers such as OpenAI, Gemini, Mistral and Voyage for RAG and semantic search.

    30k GitHub stars~604 tokensUpdated yesterday
    AI & LLM EngineeringAuto-check passed
  • Megatron-LM Base Image Bump

    NVIDIA/Megatron-LM

    Official

    Moves Megatron-LM CI to a newer NVIDIA PyTorch base image, updating both the GitHub and GitLab pins together and handling the CI follow-up.

    18k GitHub stars~2.8k tokensUpdated today
    DevOps & CloudAuto-check passed
  • Remove bracketed NemoClaw tags from GitHub issue and PR titles.

    23k GitHub stars~693 tokensUpdated today
    Marketing & SEOAuto-check passed

More from ZJLi2013/awesome-kernel-skills

All 12 skills in this repo
  • Cross Entropy Kernel

    ZJLi2013/awesome-kernel-skills

    Optimize fused cross-entropy loss kernels in Triton for NVIDIA and AMD GPUs.

    102 GitHub stars~461 tokensUpdated 6 mo ago
    Auto-check passed
  • Flash Attention Kernel

    ZJLi2013/awesome-kernel-skills

    Optimize FlashAttention-style fused attention kernels in Triton for NVIDIA and AMD GPUs.

    102 GitHub stars~697 tokensUpdated 6 mo ago
    Auto-check passed
  • Gemm Kernel Optimization

    ZJLi2013/awesome-kernel-skills

    Optimize dense matrix multiplication (GEMM) kernels in Triton for NVIDIA and AMD GPUs.

    102 GitHub stars~1.1k tokensUpdated 6 mo ago
    Auto-check passed
  • Iterative Kernel Optimization Loop

    ZJLi2013/awesome-kernel-skills

    Orchestrates continuous kernel optimization by chaining profiling, bottleneck diagnosis, tier-based optimization, verification, and benchmarking into an iterative loop.

    102 GitHub stars~2.5k tokensUpdated 6 mo ago
    Auto-check passed
  • Kernel Benchmark

    ZJLi2013/awesome-kernel-skills

    Unified kernel benchmarking protocol producing JSON results with latency, TFLOPS, GBps, and comparison against PyTorch baselines.

    102 GitHub stars~567 tokensUpdated 6 mo ago
    Auto-check passed
  • Kernel Profiling

    ZJLi2013/awesome-kernel-skills

    Profile GPU kernels using NCU (NVIDIA) or rocprof (AMD) to collect performance metrics.

    102 GitHub stars~696 tokensUpdated 6 mo ago
    Auto-check passed

Questions about Fused Moe Kernel

What does Fused Moe Kernel do?

Optimize Fused Mixture-of-Experts (MoE) kernels in Triton for NVIDIA and AMD GPUs. Fused Moe Kernel is an agent skill from ZJLi2013/awesome-kernel-skills. Optimize Fused Mixture-of-Experts (MoE) kernels in Triton for NVIDIA and AMD GPUs.

When should I use Fused Moe Kernel?

Fused Moe Kernel fits situations like: optimizing MoE layers; grouped matrix multiplication.

How do I install Fused Moe Kernel in Claude Code?

Run `npx skills add ZJLi2013/awesome-kernel-skills --skill fused-moe-kernel -a claude-code`. Or copy the skill folder (skills/kernels/fused-moe in ZJLi2013/awesome-kernel-skills) into .claude/skills/fused-moe-kernel in your project. Claude Code loads it when a task matches its description.

How do I install Fused Moe Kernel in Codex?

Run `npx skills add ZJLi2013/awesome-kernel-skills --skill fused-moe-kernel -a codex`. Or copy the skill folder (skills/kernels/fused-moe in ZJLi2013/awesome-kernel-skills) into .agents/skills/fused-moe-kernel in your project. Codex loads it when a task matches its description.

Can I use Fused Moe Kernel in Cursor, Gemini CLI or GitHub Copilot?

Cursor, Gemini CLI, GitHub Copilot and OpenCode also load SKILL.md folders. With the skills CLI, run `npx skills add ZJLi2013/awesome-kernel-skills --skill fused-moe-kernel -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/fused-moe-kernel, .gemini/skills/fused-moe-kernel, .github/skills/fused-moe-kernel and .opencode/skills/fused-moe-kernel in your project.

What does Fused Moe Kernel need to run?

Going by SKILL.md and its folder, Fused Moe Kernel needs Python for the scripts in its folder and the command-line tools its instructions call (python). Our summary lists: Python 3.

Does Fused Moe Kernel access the network?

SKILL.md names 3 domains. As links in the text: pytorch.org, github.com and emergentmind.com. This is read from the text; nothing was executed.

Is Fused Moe Kernel safe to install?

Our automated static check of SKILL.md found no risky patterns, such as piping downloads into a shell, reading credential files or hidden Unicode. It is not a guarantee. Review the folder before installing.

What licence does Fused Moe Kernel use?

No licence was found for Fused Moe Kernel or its repository. Without one, default copyright applies: ask the author before reusing or redistributing it.

How many tokens does Fused Moe Kernel use?

About 699 tokens (SKILL.md is roughly 2.8k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full.

What are the alternatives to Fused Moe Kernel?

Skills that share tags, products or a category with Fused Moe Kernel: LLM Torch Profiler Analysis (sgl-project/sglang, 37k stars), Skill Inspector (NVIDIA/SkillSpector, 20k stars), Megatron-LM Container and Dependency Setup (NVIDIA/Megatron-LM, 18k stars) and Embeddings via 9Router (decolua/9router, 30k stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.

Who maintains Fused Moe Kernel?

ZJLi2013 (a GitHub user) maintains it in ZJLi2013/awesome-kernel-skills, which has 102 GitHub stars. The repository holds 12 skills in this directory. The repository was last updated on March 31, 2026.

Source: ZJLi2013/awesome-kernel-skills on GitHub. Facts on this page come from the repository at the commit we read; the author's words are quoted as theirs.