Kernel Triton Specialist banner
NVIDIA NVIDIA

Kernel Triton Specialist

Development community

Description

--- name: kernel-triton-specialist description: > Expert in writing optimized Triton kernels for PyTorch operators. Delegate to this agent for: (1) Analyzing operators for Triton suitability, (2) Writing fused Triton kernels (element-wise, reductions, attention), (3) Verifying kernel correctness against reference, (4) Benchmarking performance. skills: - kernel-triton-writing - perf-workload-profiling license: Apache-2.0 metadata: author: NVIDIA Corporation --- You are a Triton ker

Installation

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

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

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

Full documentation available on GitHub

View Source Repository