NVIDIA

Kernel Cuda Specialist — Development agent for Claude Code

Development community

name: kernel-cuda-specialist description: Hand-writes raw CUDA C/C++ code (.cu files) with pybind11 bindings (_binding.cpp) to build custom PyTorch C++ extensions.

How to install Kernel Cuda Specialist

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

Terminal

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

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

What Kernel Cuda Specialist does

name: kernel-cuda-specialist description: > Hand-writes raw CUDA C/C++ code (.cu files) with pybind11 bindings (_binding.cpp) to build custom PyTorch C++ extensions. Delegate ONLY when the user explicitly asks to write .cu/.cpp files compiled via torch.utils.cpp_extension. Do NOT delegate for: Triton, TileIR, or any other kernel DSL/framework. NOT for CUDA Graphs, nsight profiling, torch.compile, memory analysis, or distributed training. license: Apache-2.0 metadata: author: NV

Alternatives in Development

GPU Kernel Baseline 95 ★
Ascend Kernel Developer — Ascend kernel 开发专家 Agent，通过 TileLang 设计表达和 AscendC 落地完成算子优化任务 39 ★
Strategy Critic — PROACTIVELY use this subagent immediately after the user writes or revises any strategy artifact: Strategy Blo 24 ★