구글, TPU 개발 가속화 위한 Gemini 기반 AI 에이전트 도구 공개

구글이 Google Cloud TPU에서의 머신러닝 개발을 가속화하기 위한 AI 도구 모음 Accelerator Agents를 공개했다. 이 프로젝트는 Gemini 기반의 전문화된 AI 에이전트를 활용하여 코드 마이그레이션, 커널 최적화, 성능 튜닝을 자동화한다.

주요 구성 요소

프로젝트는 두 가지 핵심 에이전트로 구성된다.

• MaxCode: 기존 PyTorch 모델과 코드베이스를 JAX로 변환한다. MaxText 프레임워크와 호환되는 코드를 생성하며, 개발자가 초안을 검토하는 Human-in-the-Loop 방식을 채택했다.
• MaxKernel: TPU용 고성능 커널 개발을 지원한다. JAX의 커널 언어인 Pallas 작성과 디버깅을 보조하며, CUDA/GPU 커널의 포팅 및 MFU(Model FLOPs Utilization) 개선을 위한 최적화 제안을 제공한다.

사용 환경 및 조건

실행을 위해 Google Cloud VM과 Python 3.11+ 환경이 필요하다. MaxCode는 CPU-only VM에서 구동 가능하지만, MaxKernel은 TPU VM을 권장한다. Agent 추론을 위해 Gemini API 접근 권한이 필수적이다.

라이선스 및 상태

이 프로젝트는 Apache License 2.0을 따르며, 현재 활발한 개발 단계에 있다. 다만 공식적으로 지원되는 Google 제품은 아니며, Google Open Source Software Vulnerability Rewards Program 대상에는 포함되지 않는다.