MaxKernel은 TPU용 고성능 커스텀 커널 개발을 위한 멀티에이전트 시스템으로, 협업적 단계별 설계를 지원하는 휴먼인더루프(HITL) 에이전트, 지표·트레이스 기반 최적화를 완전 자동화하는 자율(Auto) 에이전트, 설계 공간을 전역 탐색하도록 자율 에이전트를 확장한 그래프 기반 자율 탐색 등 세 가지 패러다임을 구현했다. 세 방식 모두 계획, 구현, 자가 디버깅, 테스트, 하드웨어 프로파일링을 담당하는 전문 서브에이전트 풀을 공유한다. TPU용 50개 다양한 커널 과제로 구성된 JaxBench와 최신 오픈소스 모델의 실제 워크로드로 평가한 결과, MaxKernel은 전문가가 직접 튜닝한 베이스라인에 필적하는 고도로 최적화된 구현을 일관되게 생성하며 유의미한 성능 향상을 보였다. 이 에이전트는 오픈소스로 공개됐다.

• •TPU 커스텀 커널 개발을 위한 멀티에이전트 시스템 MaxKernel, 세 가지 자동화 패러다임(HITL·자율·그래프 기반 탐색) 구현
• •계획·구현·자가 디버깅·테스트·하드웨어 프로파일링을 담당하는 전문 서브에이전트 풀 공유
• •50개 TPU 커널 과제 벤치마크(JaxBench)와 실제 워크로드에서 전문가 수준 튜닝에 필적하는 성능 달성
• •코드는 오픈소스로 GitHub에 공개됨