피드로 돌아가기
Custom Kernels for All from Codex and Claude
Hugging Face BlogHugging Face Blog
Backend

HuggingFace가 LLM 에이전트용 CUDA 커널 스킬을 개발해 Claude와 Codex가 transformers·diffusers 파이프라인 대상 프로덕션급 CUDA 커널을 자동 생성

Custom Kernels for All from Codex and Claude

2026년 2월 13일12intermediate

Context

CUDA 커널 작성은 H100·A100·T4별 메모리 접근 패턴·벡터화 전략·warp shuffle 감소 등 GPU 아키텍처별 최적화와 transformers·diffusers의 모듈 계층·정규화 규칙·통합 패턴 학습이 필요하며, 이는 문서와 Stack Overflow에 분산된 도메인 지식으로 인해 숙련된 개발자도 통합 실수를 범한다.

Technical Solution

  • 도메인 전문성을 구조화된 스킬로 패키징: 550토큰 규모의 SKILL.md에 H100·A100·T4 최적화 가이드, transformers·diffusers 통합 패턴, 커널 템플릿, 벤치마크 워크플로우를 집약
  • LLM 에이전트의 자동 코드 생성: Claude와 Codex가 스킬을 읽고 CUDA 소스·PyTorch 바인딩·build.toml·벤치마크 스크립트를 end-to-end 자동 생성
  • 설치 기반 스킬 배포: kernels skills add cuda-kernels --claude 명령으로 .claude/skills/cuda-kernels/에 스킬을 드롭하면 Claude Code와 Cursor가 자동 인식
  • Nix 기반 멀티 변형 빌드: kernel-builder Nix flake가 모든 PyTorch·CUDA 조합을 자동 빌드해 build/ 디렉토리에 배치
  • Kernel Hub를 통한 원라인 배포: get_kernel("your-org/your-kernel") 호출로 사용자의 Python·PyTorch·CUDA 버전에 맞는 사전 컴파일 바이너리 자동 다운로드

Key Takeaway

도메인 전문지식을 구조화된 스킬로 패키징하고 LLM 에이전트에 제공하면 높은 전문성이 필요한 작업의 자동화가 가능하며, 개발(스킬)과 배포(Hub)를 분리하는 설계로 일회성 커널 작성과 커뮤니티 재사용성 간 균형을 달성할 수 있다.


transformers·diffusers 기반 LLM 서비스를 개발하는 팀에서 RMSNorm·attention 같은 병목 연산의 CUDA 커널화가 필요할 때, 에이전트 스킬을 설치하고 '특정 GPU 아키텍처(H100)와 모델(Qwen3-8B) 대상 최적화 커널 생성' 프롬프트를 입력하면 사전 검증된 템플릿 기반 커널·바인딩·벤치마크 코드가 자동 생성되므로 개발 시간과 통합 오류를 크게 단축할 수 있다.

원문 읽기