What problem does it solve?
This Skill provides patterns and guidance for developing optimized CUDA kernels for NVIDIA GPUs, enabling benchmarking, integration with HuggingFace Kernels Hub, and diffusers/transformers workflows.
Core Features & Use Cases
- Patterns and templates for RMSNorm, RoPE, GEGLU, AdaLN, and other CUDA kernels aimed at accelerating diffusion models and large language models.
- Benchmarking scripts and end-to-end usage demos to measure performance improvements against baselines.
- Guides for integrating with HuggingFace Kernels Hub and with diffusers pipelines (Video/Image generation) and transformers models.
- Includes example scripts, integration references, and build instructions to reproduce results locally.
Quick Start
Run the quick demo by executing scripts/benchmark_example.py to benchmark optimized versus baseline kernels.