Agent skill

Cross Entropy Kernel

by ZJLi2013 in ZJLi2013/awesome-kernel-skills

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

No licenceAuto-check passed

Install Cross Entropy Kernel

skills CLI
$ npx skills add ZJLi2013/awesome-kernel-skills --skill cross-entropy-kernel -a claude-code

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

GitHub CLI
$ gh skill install ZJLi2013/awesome-kernel-skills cross-entropy-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/cross-entropy .claude/skills/cross-entropy-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
cross-entropy-kernel
GitHub stars
102
Token cost
~461 tokens
SKILL.md length
167 words
Files
3
Skills in repo
12
Repo updated
First seen
Licence
None found

At a glance

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

  • Works in 4 steps: Load row of logits into FP32 → Find max: m = max(logits) → Compute log_sum_exp = log(sum(exp(logits… → …
  • Optimizing cross-entropy
  • SKILL.md covers Overview, Core Technique, Verification and Common Pitfalls, plus 1 more section
  • Runs Python scripts from its folder; calls python

What it does

Cross Entropy Kernel is an agent skill from ZJLi2013/awesome-kernel-skills. Optimize fused cross-entropy loss kernels in Triton for NVIDIA and AMD GPUs. Covers fused log-softmax + NLL, online log-sum-exp, and large vocabulary handling. Use when writing or optimizing cross-entropy, label-smoothed CE, or similar loss kernels.

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

It works with NVIDIA AI Platform.

When your agent uses it

  • Optimizing cross-entropy
  • Label-smoothed CE
  • Similar loss kernels

Example prompts

  • “/cross-entropy-kernel”

Requirements

  • Python 3

Workflow steps

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

  1. Load row of logits into FP32
  2. Find max: m = max(logits)
  3. Compute log_sum_exp = log(sum(exp(logits - m)))
  4. Loss = -(logits[target] - m - log_sum_exp)

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):

    • github.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

Cross Entropy Kernel loads about 461 tokens when it runs. Until then it costs about 68 tokens; SKILL.md has 167 words of instructions outside code blocks.

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

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 167 words (~461 tokens).

“Fused cross-entropy: loss = -log_softmax(logits)[target] computed in a single kernel pass per row, without materializing the full softmax output.”

— opening of SKILL.md by ZJLi2013
name
cross-entropy-kernel

Read the full SKILL.md on GitHub

Files

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

  • SKILL.md
  • test_cross_entropy.py
  • triton_template.py

Open the folder on GitHubat commit aba7662

Compare with similar skills

Cross Entropy 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.

Cross Entropy Kernel compared with similar skills
SkillStarsUsed inTokensAuto-checkLicenceRepo updated
Cross Entropy Kernel this skillZJLi2013/awesome-kernel-skills102—~461Automated 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 today
    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
  • 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
  • Fused Moe Kernel

    ZJLi2013/awesome-kernel-skills

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

    102 GitHub stars~699 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 Cross Entropy Kernel

What does Cross Entropy Kernel do?

Optimize fused cross-entropy loss kernels in Triton for NVIDIA and AMD GPUs. Cross Entropy Kernel is an agent skill from ZJLi2013/awesome-kernel-skills. Optimize fused cross-entropy loss kernels in Triton for NVIDIA and AMD GPUs.

When should I use Cross Entropy Kernel?

Cross Entropy Kernel fits situations like: optimizing cross-entropy; label-smoothed CE; similar loss kernels.

How do I install Cross Entropy Kernel in Claude Code?

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

How do I install Cross Entropy Kernel in Codex?

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

Can I use Cross Entropy 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 cross-entropy-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/cross-entropy-kernel, .gemini/skills/cross-entropy-kernel, .github/skills/cross-entropy-kernel and .opencode/skills/cross-entropy-kernel in your project.

What does Cross Entropy Kernel need to run?

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

Does Cross Entropy Kernel access the network?

SKILL.md names 1 domain. As links in the text: github.com. This is read from the text; nothing was executed.

Is Cross Entropy 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 Cross Entropy Kernel use?

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

How many tokens does Cross Entropy Kernel use?

About 461 tokens (SKILL.md is roughly 1.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 Cross Entropy Kernel?

Skills that share tags, products or a category with Cross Entropy 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 Cross Entropy 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.