카카오브레인, 기계학습 성능 라이브러리 '트라이던트' 공개

트라이던트 [카카오브레인·솔랩컴퍼니 제공]

(서울=연합뉴스) 홍국기 기자 = 카카오의 인공지능(AI) 자회사인 카카오브레인은 AI 모델의 훈련·추론 속도를 향상하는 성능 라이브러리(재사용할 수 있는 코드의 집합) '트라이던트'(Trident)를 깃허브(Github)에 공개했다고 26일 밝혔다.

깃허브는 개발자들의 온라인 오픈소스 플랫폼으로, 개별 프로필을 살펴보면 개발자들이 만들어둔 코드와 그에 대한 평가를 확인할 수 있다.

트라이던트는 오픈AI의 그래픽처리장치(GPU) 프로그래밍 언어인 '트리톤'(Triton)을 기반으로 구현한 라이브러리다.

AI 모델 개발에 필요한 연산을 최적화한 커널(Kernel·컴퓨터 운영 체제의 핵심이 되는 구성요소)로 구현하고, 메타가 개발한 기계학습 프레임워크 '파이토치'(PyTorch)와 같은 사용 방법을 제공하는 것이 특징이다.

AI 개발자는 복잡하고 어려운 GPU 커널 최적화 과정을 거치지 않고 트라이던트를 활용해 모델 학습·추론에 필요한 연산 시간을 기존 작업 대비 약 25% 단축해 AI 연구·개발 작업 효율을 높일 수 있다고 카카오브레인은 소개했다.

김일두 카카오브레인 각자 대표는 "대규모 언어 모델, 이미지 생성 모델 등의 기초 연구뿐 아니라 주요 기술 공개와 같은 AI 생태계 발전에 기여하는 활동을 꾸준히 이어 나갈 것"이라고 강조했다.

redflag@yna.co.kr

▶제보는 카톡 okjebo