כתבה
arXiv cs.AI ·
MaxKernel: ייצור קרנל סקלרלי ל-TPU
MaxKernel: Agentic Kernel Generation for TPUs
MaxKernel היא מערכת של סוכנים שמייצרת קרנלים משופרים ל-TPU. היא משתמשת בשלושה פרדיגמות שונות: HITL, Auto ו-Graph-Based Autonomous Search. MaxKernel נבחנה על JaxBench והיא הציגה תוצאות משופרות בהשוואה לבסיסים ידניים.
תקציר מקורי באנגליתarXiv:2609.04523v1 Announce Type: new Abstract: Designing and authoring high-performance custom kernels for accelerators is a complex task that requires deep hardware-level expertise. Large Language Models (LLM) can be leveraged together with real-time compiler feedback to build agentic systems for kernel generation. In this work, we present MaxKernel, a multi-agent system that implements three distinct paradigms for TPU kernel development: (1) a Human-in-the-Loop (HITL) agent for collaborative, step-by-step design; (2) an Autonomous (Auto) agent that executes a fully automated, metric/trace-driven optimization loop; and (3) a Graph-Based Autonomous Search that scales the Auto agent for global exploration of the design space. All three paradigms leverage a shared pool of specialized sub-ag
קרא במקור המקורי
arxiv.org
פתח כתבה מקורית