아마존 세이지메이커 AI, MLflow 통합으로 생성형 AI 배포 최적화 가속
아마존 웹 서비스(AWS)가 자사의 머신러닝 플랫폼인 아마존 세이지메이커 AI(Amazon SageMaker AI)와 오픈소스 플랫폼인 엠엘플로우(MLflow) 간의 통합 기능을 대폭 강화했다. AWS 머신러닝 블로그에 따르면, 이제 개발자들은 생성형 AI 모델의 벤치마킹 작업과 추론 권장 결과를 단일 인터페이스인 세이지메이커 MLflow 앱으로 직접 스트리밍할 수 있다. 이는 복잡한 생성형 AI 모델을 프로덕션 환경에 배포할 때 발생하는 데이터 사일로 현상을 해소하고, 실험 반복 주기를 단축하며, 추론 최적화 워크플로우 전반에 걸쳐 완전한 재현성을 확보하기 위한 조치다.
생성형 AI 모델을 실제 서비스에 적용하는 과정은 단순히 모델을 올리는 작업이 아니다. 적절한 GPU 인스턴스 유형을 선택하고, 서빙 컨테이너를 구성하며, 병렬 처리 전략이나 투기적 디코딩(speculative decoding)과 같은 최적화 기술을 적용하는 등 고려해야 할 변수가 매우 많다. 기존에는 이러한 최적화 과정을 수동으로 벤치마킹해야 했기에 개발자들이 인프라 관리에 과도한 시간을 할애해야 했다. 세이지메이커 AI는 엔비디아(NVIDIA)의 에이아이퍼프(AIPerf)를 활용해 실제 GPU 인프라에서 각 구성을 벤치마킹하며, 시간당 첫 토큰 생성 시간, 토큰 간 지연 시간, 처리량, 초당 요청 수 등 핵심 지표를 정밀하게 측정해 최적의 배포 구성을 제안한다.
이번 통합의 핵심은 인프라 관리의 효율성 제고에 있다. 과거에는 MLflow를 사용하기 위해 사용자가 직접 인프라를 관리해야 하는 번거로움이 있었으나, 2025년 12월 도입된 서버리스 MLflow 기능은 이러한 부담을 완전히 제거했다. 실제로 기존의 수동 인프라 관리 방식과 비교했을 때, 서버리스 환경의 세이지메이커 MLflow 앱은 생성까지 약 2분이면 충분하며 별도의 인프라 유지보수 작업이 필요하지 않다. 2026년 5월부터는 MLflow 3.10 버전을 지원하기 시작하며 생성형 AI 개발을 위한 향상된 실험 추적 기능을 제공하고 있어, 개발자들은 더욱 간소화된 환경에서 모델 성능을 개선하는 데 집중할 수 있게 됐다.
이러한 기술적 진보는 생성형 AI 모델의 배포 속도와 안정성을 동시에 확보해야 하는 기업들에게 중요한 전환점이 될 전망이다. 모델 개발자가 인프라 설정에 쏟는 시간을 줄이고 데이터 기반의 의사결정에 집중할 수 있게 됨에 따라, 전체적인 개발 생산성이 향상되는 효과를 기대할 수 있다. 특히 세이지메이커 AI가 제공하는 자동화된 추론 권장 기능은 최적화 기술을 평가하여 최적의 배포 구성을 제공하므로, 비용 절감과 성능 최적화라는 두 마리 토끼를 잡아야 하는 기업들에게 실질적인 도움을 줄 것으로 보인다.
다만 이번 통합은 세이지메이커 MLflow 앱을 중심으로 이루어지며, 자체 호스팅(self-hosted) 방식의 MLflow 추적 서버로는 결과를 스트리밍할 수 없다는 점은 유의해야 한다. AWS는 2024년 6월 MLflow와의 통합을 처음 발표한 이후 지속적으로 기능을 확장해왔으며, 이번 업데이트를 통해 생성형 AI 개발 생태계 내에서의 입지를 더욱 공고히 하고 있다. 결과적으로 이번 기능 강화는 복잡한 인프라 환경을 추상화하여 개발자가 모델의 본질적인 성능 향상에만 몰입할 수 있는 환경을 조성하는 데 초점을 맞추고 있다.
이 이슈의 흐름
생성형 AI 모델을 프로덕션 환경에 배포하는 과정은 GPU 인스턴스 유형 선택, 최적화 기술 적용, 수동 벤치마킹 등 복잡하고 시간이 많이 소요되는 작업이다. 이로 인해 모델 개발자들은 인프라 관리에 많은 시간을 할애하게 되어 모델 정확도 향상에 집중하기 어렵다. Amazon SageMaker AI의 MLflow 통합은 이러한 문제를 해결하기 위해 도입되었다. 이 통합은 생성형 AI 모델의 추론 성능을 최적화하고 벤치마킹하는 과정을 자동화하여, 개발자들이 데이터 기반의 최적화 및 벤치마킹을 통해 신속하게 모델을 배포할 수 있도록 돕는다. 또한, 모든 실험 결과를 단일 플랫폼에서 추적하고 비교할 수 있게 함으로써 데이터 사일로를 줄이고, 반복 주기를 가속화하며, 워크플로우의 재현성을 높여 전반적인 개발 효율성을 향상시킨다.
- AWS, 생성형 UI 표준 AG-UI 도입…AI 에이전트 개발 속도 높인다 AWS ML Blog · 07/01
- 아마존 베드록, 미니맥스 모델 탑재…기업용 AI 에이전트 시장 공략 AWS ML Blog · 07/08
- AWS, 미 정부 전용 클라우드서 오픈AI·엔비디아 AI 모델 지원 AWS ML Blog · 07/02
- AWS, 에이전트 기억 검색 정확도 높이는 메타데이터 필터링 기능 도입 AWS ML Blog · 07/02
- 생성형 AI 프로덕션 전환의 핵심, 아마존 베드록의 탄력성 패턴 구현 AWS ML Blog · 07/01