flashinfer-aiflashinfer-aiOfficialยท4 Agent Skills Included

flashinfer

High-performance GPU kernels for fast LLM inference serving

Accelerates large language model inference with optimized GPU kernels for attention, GEMM, and mixture-of-experts operations. Eliminates slow token generation and memory bottlenecks across NVIDIA GPUs from Turing to Blackwell. Helps engineers benchmark, debug, and add custom CUDA kernels with just-in-time compilation and no manual rebuilds.
npx skills add flashinfer-ai/flashinfer --all -g -y
Available:

Instructs the AI agent on how to build, test, benchmark, and review FlashInfer GPU kernel code, including JIT compilation rules and architecture-specific constraints.

All Skills in This Repository (4)

Pure Emerald Level Indicators

Frequently Asked Questions

FAQPage Schema
How to install FlashInfer?โ–ผ

Run `npx skills add flashinfer-ai/flashinfer --all -g -y` in your terminal to install all skills in this suite globally.

What does FlashInfer do?โ–ผ

FlashInfer provides high-performance GPU kernels for LLM inference, covering attention, matrix multiplication, and mixture-of-experts operations. It powers serving frameworks like vLLM, SGLang, and TensorRT-LLM.

Which GPUs does FlashInfer support?โ–ผ

It supports NVIDIA GPUs from Turing (SM75) through Blackwell (SM120), including A100, H100, B200, and RTX series cards.

How to debug CUDA crashes in FlashInfer?โ–ผ

Enable API logging with the FLASHINFER_LOGLEVEL environment variable to capture tensor shapes and values before a crash. The debug skill guides your agent through diagnosing memory errors and NaN issues.

Can I add custom CUDA kernels with AI help?โ–ผ

Yes. The included skills walk your AI agent through writing, binding, and testing new CUDA kernels using FlashInfer's JIT compilation system.

Related Repositories in Software Engineering

View All in Software Engineeringโ†’