softmax-kernel-optimization

softmax-kernel-optimization is a skill for Claude Code, Codex from ZJLi2013/awesome-kernel-skills. It costs 59 tokens per session (803 once invoked), scanned A, original, no licence file.

A guide to optimizing fused softmax kernels in Triton, where softmax converts a row of values into normalized weights. It covers stable maximum and sum calculations, multiple-row processing, and warp-level work.

In plain words
What is it for?
Use it when writing or tuning softmax, log-softmax, or similar row-wise normalization kernels for NVIDIA or AMD GPUs.
Why use it?
It helps improve softmax speed while avoiding numerical errors caused by very large or very small values.

Skill for Claude CodeCodex

Which agent this was written for is unclear — body not stored (licence); the path alone says nothing.

Good fit Use it when writing or tuning softmax, log-softmax, or similar row-wise normalization kernels for NVIDIA or AMD GPUs.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/zjli2013/awesome-kernel-skills/softmax
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add ZJLi2013/awesome-kernel-skills --skill softmax
Clone the repo
git clone --depth 1 https://github.com/ZJLi2013/awesome-kernel-skills

Made for: Claude Code, Codex.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for softmax-kernel-optimization

README.md
[![agentmods](https://agentmods.dev/badge/skills/zjli2013/awesome-kernel-skills/softmax/github.svg)](https://agentmods.dev/skills/zjli2013/awesome-kernel-skills/softmax)
Your own site
<a href="https://agentmods.dev/skills/zjli2013/awesome-kernel-skills/softmax"><img src="https://agentmods.dev/badge/skills/zjli2013/awesome-kernel-skills/softmax/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for softmax-kernel-optimization

Your own site · 80×15
<a href="https://agentmods.dev/skills/zjli2013/awesome-kernel-skills/softmax"><img src="https://agentmods.dev/badge/skills/zjli2013/awesome-kernel-skills/softmax.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 59 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 803 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe.
Origin unknown No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00059 $0.00803
Opus 5 $0.00030 $0.00402
Sonnet 5 $0.00012 $0.00161
Haiku 4.5 $0.00006 $0.00080

Measured 9d ago against content hash e5d4996d08de, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-08, from the pricing page.

Security

Grade A, and why

softmax-kernel-optimization scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 9d ago.

The scan reads SKILL.md. This mod also ships 2 executable files (test_softmax.py, triton_template.py), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

skills/kernels/softmax/SKILL.md · 85 lines

The source is not reproduced here

No licence file

A repository with no LICENSE is all rights reserved by default, so the body is not copied here. The metadata, the measurements and the link are.

Read it on GitHub

Files

What ships with it

2 files beside SKILL.md in the same directory: the scripts, references and assets a skill reads on demand. Not counted in the per-session cost; read them before you install if any of them is executable.

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 9d ago First seen · 85 lines · 59 tokens per session scan A e5d4996d08de

Subscribe to this mod's changes

softmax-kernel-optimization is a skill published in the GitHub repository ZJLi2013/awesome-kernel-skills (100 stars, last pushed 5mo ago), with no licence file. It adds 59 tokens to every session and 803 once invoked, about $0.0003 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Related

Other skills, from other repositories

gke-compute-classes

Configures, optimizes, and troubleshoots GKE ComputeClasses. Use when configuring Spot VMs with on-demand fallback, targeting specific accelerators (GPUs/TPUs) or machine families, restricting ComputeClass access, or debugging pending pods related to node pool auto-creation. Do not use for cluster-level Node Auto…

google/skills · 83 tokens

jetson-diagnostic

Read-only Jetson health snapshot for identity, memory, GPU, thermal, power, storage, services, and top processes.

NVIDIA/skills · 30 tokens

doca-socket-relay

Use this skill when the operator is driving the DOCA Socket Relay to bridge a socket-oriented host application onto a BlueField DPU peer without rewriting it — picking the deployment shape (in-process, sidecar, or BlueField service container), configuring the host-side socket and the DPU-side forwarding endpoint…

NVIDIA/skills · 236 tokens

offensive-z-wave

Z-Wave attack methodology — sniffing with Z-Force / EZ-Wave / RTL-SDR + ZniffMobile, S0 (legacy) network-key derivation flaw and key reuse, S2 (modern) ECDH commissioning analysis, replay/injection on unauthenticated nodes, default-key brute-force on test deployments, and home-automation hub pivots. Use when targeting…

SnailSploit/Claude-Red · 113 tokens

hsb-flash

Flash the FPGA on an HSB board connected to an NVIDIA devkit. Supports HSB Lattice boards (FPGA versions 2407, 2412, 2507, 2510) and Leopard Imaging VB1940 "all-in-one" cameras (FPGA versions 2507, 2510). Uses release-specific YAML manifests and board-type-specific program commands. Lattice and VB1940 commands must…

NVIDIA/skills · 94 tokens

jetson-validate-image

Use after jetson-flash-image to run static BSP checks, on-target smoke/regression tests on a flashed DUT, or both. Not for build or flash steps. Triggers: validate bsp, on-target validation.

NVIDIA/skills · 50 tokens