AI 에이전트 연구
MaxKernel: Agentic Kernel Generation for TPUs
ARarXiv
하드웨어 전문가의 깊은 지식이 필요했던 TPU용 고성능 커널 설계 문제를 해결하기 위해, LLM 기반 다중 에이전트 시스템 MaxKernel을 개발했습니다.
세 줄 요약
- MaxKernel은 인간 협업형(HITL), 자율 최적화 루프, 그래프 검색 등 3가지 패러다임을 통해 커널을 생성하며 전문가 수준의 성능을 입증했습니다.
- 전문 지식 장벽을 낮추고 하드웨어 최적화를 자동화함으로써, 누구나 효율적인 고성능 컴퓨팅 환경 구축이 가능해집니다.
- 계획, 디버깅 등 전문 서브 에이전트를 공유하는 구조를 가졌으며, 현재 시스템 코드가 오픈소스로 공개되어 활용할 수 있습니다.
하드웨어 전문가의 깊은 지식이 필요했던 TPU용 고성능 커널 설계 문제를 해결하기 위해, LLM 기반 다중 에이전트 시스템 MaxKernel을 개발했습니다.