Kernel Cute Specialist banner
NVIDIA NVIDIA

Kernel Cute Specialist

Development community

Description

--- name: kernel-cute-specialist description: > Expert in writing optimized GPU kernels using CuTe DSL for NVIDIA GPUs (Ampere through Blackwell). This is the CuTe DSL specialist -- NOT Triton. CuTe DSL uses cute.jit/cute.kernel decorators and cutlass.cute imports; Triton uses triton.jit and tl.* primitives -- they are completely different frameworks. Delegate to this agent for: (1) Writing CuTe DSL kernels from CUTLASS examples or element-wise patterns, (2) Lowering PyTorch operator

Installation

Installs to ~/.claude/agents/nvidia-tensorrt-llm-kernel-cute-specialist.md

Terminal
mkdir -p ~/.claude/agents && curl -fsSL https://raw.githubusercontent.com/NVIDIA/TensorRT-LLM/HEAD/.claude/agents/kernel-cute-specialist.md -o ~/.claude/agents/nvidia-tensorrt-llm-kernel-cute-specialist.md

Restart Claude Code, or start a new session, for it to be picked up.

Full documentation available on GitHub

View Source Repository